diff --git a/modules/deeplearning/cloud_opd_srcnn_viirs.py b/modules/deeplearning/cloud_opd_srcnn_viirs.py
index 9db4b165502867b5308c835fedc7bfcda17d708e..913b195fec835b86bdb566e08f7c0671f5f813ed 100644
--- a/modules/deeplearning/cloud_opd_srcnn_viirs.py
+++ b/modules/deeplearning/cloud_opd_srcnn_viirs.py
@@ -694,6 +694,7 @@ def run_evaluate_static(in_file, out_file, ckpt_dir):
     bt = get_grid_values_all(h5f, 'orig/temp_11_0um')
     bt = np.where(np.isnan(bt), 0, bt)
     bt = bt[nn.slc_y_m, nn.slc_x_m]
+    bt = np.expand_dims(bt, axis=0)
     bt = nn.upsample(bt)
     bt = normalize(bt, 'temp_11_0um_nom', mean_std_dct)
 
@@ -710,7 +711,8 @@ def run_evaluate_static(in_file, out_file, ckpt_dir):
 
     cld_opd = get_grid_values_all(h5f, 'orig/'+label_param)
     cld_opd = np.where(np.isnan(cld_opd), 0, cld_opd)
-    cld_opd = cld_opd[:, nn.slc_y_2, nn.slc_x_2]
+    cld_opd = cld_opd[nn.slc_y_2, nn.slc_x_2]
+    cld_opd = np.expand_dims(cld_opd, axis=0)
     cld_opd = nn.upsample(cld_opd)
     cld_opd = normalize(cld_opd, label_param, mean_std_dct)