diff --git a/modules/deeplearning/cloud_opd_srcnn_abi.py b/modules/deeplearning/cloud_opd_srcnn_abi.py
index 83198e71e3ec65cca5f7201629b37837921c1498..0fb761bdc86831c053471fa2e2bd0638f8727cb2 100644
--- a/modules/deeplearning/cloud_opd_srcnn_abi.py
+++ b/modules/deeplearning/cloud_opd_srcnn_abi.py
@@ -698,9 +698,9 @@ class SRCNN:
         self.build_training()
         self.build_evaluation()
 
-        # ckpt = tf.train.Checkpoint(step=tf.Variable(1), model=self.model)
-        # ckpt_manager = tf.train.CheckpointManager(ckpt, ckpt_dir, max_to_keep=3)
-        # ckpt.restore(ckpt_manager.latest_checkpoint)
+        ckpt = tf.train.Checkpoint(step=tf.Variable(1), model=self.model)
+        ckpt_manager = tf.train.CheckpointManager(ckpt, ckpt_dir, max_to_keep=3)
+        ckpt.restore(ckpt_manager.latest_checkpoint)
 
     def do_inference(self, inputs):
         self.reset_test_metrics()
@@ -740,7 +740,7 @@ class SRCNN:
         t = np.arange(0, (LEN_X - 16) + 4, 0.5)
         s = np.arange(0, (LEN_Y - 16) + 4, 0.5)
 
-        # refl = np.where(np.isnan(refl), 0, bt)
+        # refl = np.where(np.isnan(refl), 0, refl)
         # refl = refl[slc_y, slc_x]
         # refl = np.expand_dims(refl, axis=0)
         # refl_us = upsample_static(refl, x_2, y_2, t, s, None, None)