diff --git a/modules/deeplearning/cloud_fraction_fcn.py b/modules/deeplearning/cloud_fraction_fcn.py
index bcd94297df22deeb8e321a6533780ed09a976027..e5433cafe3a17393ba8d823b4bdf59be8c13bb00 100644
--- a/modules/deeplearning/cloud_fraction_fcn.py
+++ b/modules/deeplearning/cloud_fraction_fcn.py
@@ -761,7 +761,7 @@ def run_restore_static(directory, ckpt_dir, out_file=None):
     labels, preds = nn.run_restore(directory, ckpt_dir)
     if out_file is not None:
         np.save(out_file,
-                [np.squeeze(labels), preds.argmax(axis=3), preds[:, :, :, 0], preds[:, :, :, 1], preds[:, :, :, 2]])
+                [np.squeeze(labels), preds.argmax(axis=3)])
 
 
 def run_evaluate_static(in_file, out_file, ckpt_dir):
@@ -816,9 +816,6 @@ def analyze_3cat(file):
     tup = np.load(file, allow_pickle=True)
     lbls = tup[0]
     pred = tup[1]
-    prob_0 = tup[2]
-    prob_1 = tup[3]
-    prob_2 = tup[4]
 
     lbls = lbls.flatten()
     pred = pred.flatten()
@@ -918,9 +915,6 @@ def analyze_5cat(file='/Users/tomrink/cld_opd_frac.npy'):
     tup = np.load(file, allow_pickle=True)
     lbls = tup[0]
     pred = tup[1]
-    # prob_0 = tup[2]
-    # prob_1 = tup[3]
-    # prob_2 = tup[4]
 
     lbls = lbls.flatten()
     pred = pred.flatten()