diff --git a/python/paddle/fluid/compiler.py b/python/paddle/fluid/compiler.py
index c568f9d2546f3684a27486c8b46b93be3527b9ee..4441481093fb5068b770d7f739424bfb1609e5a0 100644
--- a/python/paddle/fluid/compiler.py
+++ b/python/paddle/fluid/compiler.py
@@ -37,7 +37,7 @@ def _place_obj(place):
 
 def _is_pserver_mode(main_program):
     main = main_program if main_program \
-        else default_main_program()
+        else framework.default_main_program()
     for op in main.global_block().ops:
         if op.type in ["send", "recv"]:
             return True