diff --git a/modules/deeplearning/cloud_opd_srcnn_viirs.py b/modules/deeplearning/cloud_opd_srcnn_viirs.py
index 60c5cb23a326567332b3118813bd7a906082de19..6dc8c39f3fac26bc07a15f190a6a76d6eb712244 100644
--- a/modules/deeplearning/cloud_opd_srcnn_viirs.py
+++ b/modules/deeplearning/cloud_opd_srcnn_viirs.py
@@ -719,8 +719,9 @@ def run_evaluate_static(in_file, out_file, ckpt_dir):
     cld_opd = nn.upsample(cld_opd)
     cld_opd = normalize(cld_opd, label_param, mean_std_dct)
 
-    data = np.stack([bt, refl, cld_opd], axis=2)
-    data = np.expand_dims(data, axis=0)
+    data = np.stack([bt, refl, cld_opd], axis=3)
+    print(data.shape)
+    #data = np.expand_dims(data, axis=0)
 
     h5f.close()