diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py
index 1752ead34820b61927536f2f5f30178e34771dbc..fe437eea204ee8e6328c365d4ffc2231c68f699b 100644
--- a/modules/deeplearning/srcnn_l1b_l2.py
+++ b/modules/deeplearning/srcnn_l1b_l2.py
@@ -276,7 +276,7 @@ class SRCNN:
                 tmp = np.where(tmp > 1.0, 1.0, tmp)
             tmp = np.where(np.isnan(tmp), 0, tmp)
         tmp = resample_2d_linear(x_2, y_2, tmp, t, s)
-        tmp = tmp[y_k, x_k]
+        tmp = tmp[:, y_k, x_k]
         data_norm.append(tmp)
         # ---------
         data = np.stack(data_norm, axis=3)