diff --git a/modules/deeplearning/cloud_opd_srcnn_abi.py b/modules/deeplearning/cloud_opd_srcnn_abi.py
index 67bbdbc00caab8fe1a3582595150a29bc2611fae..5049c9bc5c68611144cb376d723fe383c402b228 100644
--- a/modules/deeplearning/cloud_opd_srcnn_abi.py
+++ b/modules/deeplearning/cloud_opd_srcnn_abi.py
@@ -928,7 +928,7 @@ class SRCNN:
         cld_opd_sres = self.do_inference(data)
         cld_opd_sres = denormalize(cld_opd_sres, label_param, mean_std_dct)
 
-        return cld_opd_sres, bt_us.shape[1], bt_us.shape[2]
+        return cld_opd_sres, cld_opd_us.shape[1], cld_opd_us.shape[2]
 
 
 def run_restore_static(directory, ckpt_dir, out_file=None):