diff --git a/tests/flytekit/unit/core/test_type_engine.py b/tests/flytekit/unit/core/test_type_engine.py
index 9bdae71b6a0..465317b5ebd 100644
--- a/tests/flytekit/unit/core/test_type_engine.py
+++ b/tests/flytekit/unit/core/test_type_engine.py
@@ -2275,8 +2275,12 @@ def t1(a: int) -> int:
         downstream_t(a=v, df=df)
 
         return v_1
+    
+    @workflow
+    def wf(a: int) -> int:
+        return t1(a=a)
 
-    assert t1(a=3) == 9
+    assert wf(a=3) == 9
 
 
 def test_literal_hash_int_can_be_set():