diff --git a/src/django_grainy/handlers.py b/src/django_grainy/handlers.py
index cce441c..31fbaa8 100644
--- a/src/django_grainy/handlers.py
+++ b/src/django_grainy/handlers.py
@@ -80,9 +80,8 @@ def set_namespace_base(cls, value: Namespace):
             raise TypeError("`value` needs to be a Namespace instance")
         cls.namespace_base = value
 
-    # TODO: type hint
     @classmethod
-    def set_parent(cls, parent):
+    def set_parent(cls, parent: Model):
         cls.parent = parent