diff --git a/modules/deeplearning/cloud_fraction_fcn_viirs.py b/modules/deeplearning/cloud_fraction_fcn_viirs.py
index 76941cf54fb832cca69d6bc577571b5c958b7c5d..b45c13d2af68ac864e20902a24e18d9dc131730a 100644
--- a/modules/deeplearning/cloud_fraction_fcn_viirs.py
+++ b/modules/deeplearning/cloud_fraction_fcn_viirs.py
@@ -760,7 +760,9 @@ def run_restore_static(directory, ckpt_dir, out_file=None):
     labels, preds, inputs = nn.run_restore(directory, ckpt_dir)
     if out_file is not None:
         np.save(out_file,
-                [np.squeeze(labels), preds.argmax(axis=3), inputs[:, 1:65, 1:65, 4]])
+                [np.squeeze(labels), preds.argmax(axis=3),
+                 inputs[:, 1:65, 1:65, 0], inputs[:, 1:65, 1:65, 1],
+                 inputs[:, 1:65, 1:65, 2], inputs[:, 1:65, 1:65, 3], inputs[:, 1:65, 1:65, 4]])
 
 
 def run_evaluate_static(in_file, out_file, ckpt_dir):