diff --git a/python/paddle/static/__init__.py b/python/paddle/static/__init__.py
index 89da75ae91e40ed5fcf8d45dfca4e2628cc24f9e..688bff4a678f2af224a5c541eb2331569edb717e 100644
--- a/python/paddle/static/__init__.py
+++ b/python/paddle/static/__init__.py
@@ -85,6 +85,7 @@ __all__ = [     #noqa
            'load',
            'save_inference_model',
            'load_inference_model',
+           'normalize_program',
            'load_program_state',
            'set_program_state',
            'cpu_places',
diff --git a/python/paddle/static/io.py b/python/paddle/static/io.py
index 58e8ebc481d799955d8a738e4c8a581ccd319679..a9cae0c14e3b19f78e08f8069e8e33bd68077a28 100644
--- a/python/paddle/static/io.py
+++ b/python/paddle/static/io.py
@@ -157,7 +157,7 @@ def normalize_program(program, feed_vars, fetch_vars):
             exe.run(paddle.static.default_startup_program())
 
             # normalize main program.
-            program = default_main_program()
+            program = paddle.static.default_main_program()
             normalized_program = paddle.static.normalize_program(program, [image], [predict])
 
     """