diff --git a/tools/infer.py b/tools/infer.py
index d33d1e3ebecaa342643817c1302220f202d93885..64049e3fc2b513d6e1e8946edc4bb33518ea8f40 100644
--- a/tools/infer.py
+++ b/tools/infer.py
@@ -106,7 +106,7 @@ def prune_feed_vars(feeded_var_names, target_vars, prog):
     """
     exist_var_names = []
     prog = prog.clone()
-    prog = prog._prune(feeded_var_names, targets=target_vars)
+    prog = prog._prune(targets=target_vars)
     global_block = prog.global_block()
     for name in feeded_var_names:
         try: