diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py
index 29ef22b6f1770ac21831b54c39f254721836086d..5e78a128fda335390ef262a8997e1254d48bd904 100644
--- a/modules/deeplearning/espcn.py
+++ b/modules/deeplearning/espcn.py
@@ -605,8 +605,8 @@ class ESPCN:
 
         self.reset_test_metrics()
 
-        for data0, data1, label in self.test_dataset:
-            ds = tf.data.Dataset.from_tensor_slices((data0, data1, label))
+        for data, label in self.test_dataset:
+            ds = tf.data.Dataset.from_tensor_slices((data, label))
             ds = ds.batch(BATCH_SIZE)
             for mini_batch_test in ds:
                 self.predict(mini_batch_test)
@@ -639,7 +639,7 @@ class ESPCN:
 
         self.reset_test_metrics()
 
-        pred = self.model([data])
+        pred = self.model([data], training=False)
         self.test_probs = pred
         pred = pred.numpy()