diff --git a/predict.py b/predict.py
index 296fdb8..fdb95ea 100755
--- a/predict.py
+++ b/predict.py
@@ -10,8 +10,7 @@ from torchvision import transforms
 
 from utils.data_loading import BasicDataset
 from unet import UNet
-from utils import plot_img_and_mask
-
+from utils.utils import plot_img_and_mask
 
 def predict_img(net,
                 full_img,