diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py
index db1f7aad21d1b9ef4ac024588753d90bcef0f7f6..60d89f0dd60c5368be55590c7ca0a1db6dc33163 100644
--- a/modules/deeplearning/srcnn_l1b_l2.py
+++ b/modules/deeplearning/srcnn_l1b_l2.py
@@ -652,6 +652,11 @@ class SRCNN:
         return self.do_evaluate(data, ckpt_dir)
 
 
+def run_restore_static(directory, ckpt_dir):
+    nn = SRCNN()
+    nn.run_restore(directory, ckpt_dir)
+
+
 def run_evaluate_static(in_file, out_file, ckpt_dir):
     h5f = h5py.File(in_file, 'r')
     grd_a = get_grid_values_all(h5f, data_params[0])
@@ -700,7 +705,7 @@ def analyze():
 
     h5f.close()
 
-    return grd, grd_hr
+    return grd, grd_lr, grd_hr
 
 
 if __name__ == "__main__":