diff --git a/nvfuser/__init__.py b/nvfuser/__init__.py
index f4adc81c5b0..7ae03ead3d6 100644
--- a/nvfuser/__init__.py
+++ b/nvfuser/__init__.py
@@ -372,7 +372,8 @@ def execute(
                 _enable_options=_enable_options,
                 _disable_options=_disable_options,
             )
-            out_tensors = []
+
+            out_tensors: list[torch.Tensor] = []
             for out_dtensor in out_dtensors:
                 if out_dtensor.mesh.size == 0:
                     out_tensors.append(out_dtensor.local)