diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index 78e551d7ddb6722174ee06b9f0bce8dc1a851f8f..2fd21c811d22bd556b4664a764417afb74ecd1a2 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -924,8 +924,8 @@ class IcingIntensityNN:
         self.test_loss.reset_states()
         self.test_accuracy.reset_states()
 
-        for data0, label in self.test_dataset:
-            ds = tf.data.Dataset.from_tensor_slices((data0, label))
+        for data0, data1, label in self.test_dataset:
+            ds = tf.data.Dataset.from_tensor_slices((data0, data1, label))
             ds = ds.batch(BATCH_SIZE)
             for mini_batch_test in ds:
                 self.predict(mini_batch_test)