diff --git a/python/paddle/fluid/layers/math_op_patch.py b/python/paddle/fluid/layers/math_op_patch.py
index be7bf69c30222fa1d07742060547ec52005b4afd..73ae377ac346f3a2e6e87af479dfd423751b9304 100644
--- a/python/paddle/fluid/layers/math_op_patch.py
+++ b/python/paddle/fluid/layers/math_op_patch.py
@@ -42,8 +42,7 @@ def monkey_patch_variable():
     def current_block(var):
         if in_dygraph_mode():
             return default_main_program().global_block()
-        else:
-            return var.block
+        return var.block.program.current_block()
 
     def create_new_tmp_var(block, dtype):
         tmp_name = unique_tmp_name()