diff --git a/utils.py b/utils.py
index e3bdad0..245d29c 100644
--- a/utils.py
+++ b/utils.py
@@ -58,7 +58,8 @@ def preprocess(path, scale=3):
   (width, height) = scaled_image.size
   input_ = np.array(list(scaled_image.getdata())).astype(np.float).reshape((height, width))
 
-  return input_, label_
+  # I think the cropped_image should be returned as cropped image, but I do not tested.
+  return input_, cropped_image
 
 def prepare_data(sess, dataset):
   """