diff --git a/paddle/fluid/eager/to_static/run_program_impl.cc b/paddle/fluid/eager/to_static/run_program_impl.cc
index c54f4c9d386c59..9e4011e8080519 100644
--- a/paddle/fluid/eager/to_static/run_program_impl.cc
+++ b/paddle/fluid/eager/to_static/run_program_impl.cc
@@ -573,7 +573,7 @@ std::vector<paddle::Tensor> RunProgramImpl(
 #endif
 
     auto passed_kernel_program = paddle::framework::ApplyIrPass(
-        forward_program.get(), place, no_need_buffer_name_set);
+        program.get(), place, no_need_buffer_name_set);
     const auto &new_block = passed_kernel_program->block();
     passed_kernel_program = paddle::framework::ApplyRemoveShadowFeedPass(
         std::move(passed_kernel_program), new_block, place, global_inner_scope);