diff --git a/modules/deeplearning/cloud_opd_srcnn_abi_v2.py b/modules/deeplearning/cloud_opd_srcnn_abi_v2.py
index 5e71cccc128a69b4b309a4ef711a449e820c9d17..def9bfd8ca411eac0fbb24cb037aafb8a098854d 100644
--- a/modules/deeplearning/cloud_opd_srcnn_abi_v2.py
+++ b/modules/deeplearning/cloud_opd_srcnn_abi_v2.py
@@ -811,17 +811,17 @@ class SRCNN:
 
         refl_sub_lo = np.expand_dims(refl_sub_lo, axis=0)
         refl_sub_lo = upsample_nearest(refl_sub_lo)
-        refl_sub_lo = refl_sub_lo[self.slc_y, self.slc_x]
+        refl_sub_lo = refl_sub_lo[:, self.slc_y, self.slc_x]
         refl_sub_lo = normalize(refl_sub_lo, 'refl_0_65um_nom', mean_std_dct)
 
         refl_sub_hi = np.expand_dims(refl_sub_hi, axis=0)
         refl_sub_hi = upsample_nearest(refl_sub_hi)
-        refl_sub_hi = refl_sub_hi[self.slc_y, self.slc_x]
+        refl_sub_hi = refl_sub_hi[:, self.slc_y, self.slc_x]
         refl_sub_hi = normalize(refl_sub_hi, 'refl_0_65um_nom', mean_std_dct)
 
         refl_sub_std = np.expand_dims(refl_sub_std, axis=0)
         refl_sub_std = upsample_nearest(refl_sub_std)
-        refl_sub_std = refl_sub_std[self.slc_y, self.slc_x]
+        refl_sub_std = refl_sub_std[:, self.slc_y, self.slc_x]
 
         t1 = time.time()
         print('upsample/normalize time: ', (t1 - t0))