diff --git a/backends/vulkan/test/op_tests/utils/gen_computegraph.py b/backends/vulkan/test/op_tests/utils/gen_computegraph.py
index f6ee9c78a1..8cefb5be04 100644
--- a/backends/vulkan/test/op_tests/utils/gen_computegraph.py
+++ b/backends/vulkan/test/op_tests/utils/gen_computegraph.py
@@ -448,7 +448,7 @@ def set_output(self, ref: ValueRefList, include_declarations: bool = True) -> st
         if isinstance(ref, list):
             ret_str = ""
             for r in ref[:-1]:
-                ret_str += self.set_output(r)
+                ret_str += self.set_output(r, include_declarations)
             return ret_str
         elif ref.src_cpp_type == TENSOR_VECTOR:
             assert ref.is_out