diff --git a/src/dflow/python/op.py b/src/dflow/python/op.py
index 3f910669..c09fa1f6 100644
--- a/src/dflow/python/op.py
+++ b/src/dflow/python/op.py
@@ -398,9 +398,9 @@ def from_graph(cls, graph):
             op = getattr(mod, graph["name"])
         return op
 
-    def set_output(self, name, value):
+    def set_output(self, name, value, symlink=True):
         self.outputs[name] = value
-        self.handle_outputs({name: value}, symlink=True)
+        self.handle_outputs({name: value}, symlink=symlink)
 
     def handle_outputs(self, outputs, symlink=False):
         os.makedirs("%s/outputs/parameters" % self.tmp_root, exist_ok=True)