diff --git a/t4_devkit/schema/tables/autolabel_metadata.py b/t4_devkit/schema/tables/autolabel_metadata.py
index 26bc247..ccfa359 100644
--- a/t4_devkit/schema/tables/autolabel_metadata.py
+++ b/t4_devkit/schema/tables/autolabel_metadata.py
@@ -31,11 +31,11 @@ class AutolabelModel:
     )
 
     @staticmethod
-    def to_autolabel_model(x) -> list[AutolabelModel] | None:
+    def to_autolabel_model(x: list[dict | AutolabelModel] | None) -> list[AutolabelModel] | None:
         """Convert input to a list of AutolabelModel instances.
 
         Args:
-            x: Input to convert. Can be None, a list of dicts, or a list of AutolabelModel instances.
+            x (list[dict | AutolabelModel] | None): Input to convert. Can be None, a list of dicts, or a list of AutolabelModel instances.
 
         Returns:
             list[AutolabelModel] | None: Converted list of AutolabelModel instances or None.