diff --git a/modules/deeplearning/cloud_opd_srcnn_abi_v2.py b/modules/deeplearning/cloud_opd_srcnn_abi_v2.py
index 8e568e406c58e99093bd5ff70f599bbab59f646d..b056f4efdf7eda3861ce37e9d22e1461c37056c1 100644
--- a/modules/deeplearning/cloud_opd_srcnn_abi_v2.py
+++ b/modules/deeplearning/cloud_opd_srcnn_abi_v2.py
@@ -768,7 +768,7 @@ class SRCNN:
 
         cld_opd_sres_out = np.zeros((LEN_Y, LEN_X), dtype=np.int8)
         border = int((KERNEL_SIZE - 1) / 2)
-        cld_opd_sres_out[border:LEN_Y - border, border:LEN_X - border] = cld_opd_sres[0, :, :]
+        cld_opd_sres_out[border:LEN_Y - border, border:LEN_X - border] = cld_opd_sres[0, :, :, 0]
 
         h5f.close()