diff --git a/paddle/fluid/framework/details/var_handle.h b/paddle/fluid/framework/details/var_handle.h
index ba8b38d1e7..d8c2bc40b9 100644
--- a/paddle/fluid/framework/details/var_handle.h
+++ b/paddle/fluid/framework/details/var_handle.h
@@ -57,7 +57,8 @@ struct VarHandleBase {
   void RemoveOutput(OpHandleBase* out, ir::Node* node) {
     pending_ops_.erase(out);
     node_->outputs.erase(
-        std::remove(node_->outputs.begin(), node_->outputs.end(), node));
+        std::remove(node_->outputs.begin(), node_->outputs.end(), node),
+        node_->outputs.end());
   }
 
   void ClearGeneratedOp() {