diff --git a/src/engine/src/owe_python_sdk/schema.py b/src/engine/src/owe_python_sdk/schema.py
index dd8579dd..d15c94c5 100644
--- a/src/engine/src/owe_python_sdk/schema.py
+++ b/src/engine/src/owe_python_sdk/schema.py
@@ -553,6 +553,7 @@ def backwards_compatibility_transforms(cls, values):
         return values
     
 class TemplateTask(BaseTask):
+    type: Literal["template"]
     uses: Union[str, Uses]
 
 class ApplicationTask(BaseTask):