diff --git a/client/ayon_core/plugins/publish/integrate_traits.py b/client/ayon_core/plugins/publish/integrate_traits.py
index 3fbf57be88..45f32be4a0 100644
--- a/client/ayon_core/plugins/publish/integrate_traits.py
+++ b/client/ayon_core/plugins/publish/integrate_traits.py
@@ -150,13 +150,13 @@ class TemplateItem:
     anatomy: Anatomy
     template: str
     template_data: dict[str, Any]
-    template_object: AnatomyTemplateItem
+    template_object: "AnatomyTemplateItem"
 
     def __init__(self,
-        anatomy: Anatomy,
+        anatomy: "Anatomy",
         template: str,
         template_data: dict[str, Any],
-        template_object: AnatomyTemplateItem):
+        template_object: "AnatomyTemplateItem"):
         """Initialize TemplateItem.
 
         Args: