diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index 11a559b53a552943e0d1bd301f9784dfe3ca1cc8..be27c00076206c01ea3f1ef7eb6daae71038bde6 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -161,6 +161,7 @@ class IcingIntensityNN:
 
         self.test_labels = []
         self.test_preds = []
+        self.test_probs
 
         self.learningRateSchedule = None
         self.num_data_samples = None
@@ -717,6 +718,7 @@ class IcingIntensityNN:
 
         labels = np.concatenate(self.test_labels)
         preds = np.concatenate(self.test_preds)
+        self.test_probs = preds
         preds = np.where(preds > 0.5, 1, 0)
 
         self.test_labels = labels