diff --git a/modules/deeplearning/cloud_opd_srcnn_viirs.py b/modules/deeplearning/cloud_opd_srcnn_viirs.py
index 637fc2072096bfcae050a853ca27072c17d93951..60c5cb23a326567332b3118813bd7a906082de19 100644
--- a/modules/deeplearning/cloud_opd_srcnn_viirs.py
+++ b/modules/deeplearning/cloud_opd_srcnn_viirs.py
@@ -688,11 +688,11 @@ def run_evaluate_static(in_file, out_file, ckpt_dir):
     LEN_X -= 16
     refl = np.where(np.isnan(refl), 0, refl)
     refl = normalize(refl, 'refl_0_65um_nom', mean_std_dct)
-    refl = np.expand_dims(refl, axis=0)
 
     nn = SRCNN(LEN_Y=LEN_Y, LEN_X=LEN_X)
 
     refl = refl[nn.slc_y, nn.slc_x]
+    refl = np.expand_dims(refl, axis=0)
 
     bt = get_grid_values_all(h5f, 'orig/temp_11_0um')
     bt = np.where(np.isnan(bt), 0, bt)