diff --git a/modules/deeplearning/cloud_fraction_fcn_abi.py b/modules/deeplearning/cloud_fraction_fcn_abi.py
index 87441fe1fcf3ca1bf065def075c3ba7835c72f77..0cfce99ee31ac8fbe990b6b7760e7a0c40e53071 100644
--- a/modules/deeplearning/cloud_fraction_fcn_abi.py
+++ b/modules/deeplearning/cloud_fraction_fcn_abi.py
@@ -761,11 +761,11 @@ def run_restore_static(directory, ckpt_dir, out_file=None):
     if out_file is not None:
         np.save(out_file,
                 [np.squeeze(labels), preds.argmax(axis=3),
-                 denormalize(inputs[:, 1:65, 1:65, 0], 'temp_11_0um_nom', mean_std_dct),
-                 denormalize(inputs[:, 1:65, 1:65, 1], 'refl_0_65um_nom', mean_std_dct),
-                 denormalize(inputs[:, 1:65, 1:65, 2], 'refl_0_65um_nom', mean_std_dct),
-                 inputs[:, 1:65, 1:65, 3],
-                 inputs[:, 1:65, 1:65, 4]])
+                 denormalize(inputs[:, 1:17, 1:17, 0], 'temp_11_0um_nom', mean_std_dct),
+                 denormalize(inputs[:, 1:17, 1:17, 1], 'refl_0_65um_nom', mean_std_dct),
+                 denormalize(inputs[:, 1:17, 1:17, 2], 'refl_0_65um_nom', mean_std_dct),
+                 inputs[:, 1:17, 1:17, 3],
+                 inputs[:, 1:17, 1:17, 4]])
 
 
 def run_evaluate_static(in_file, out_file, ckpt_dir):