diff --git a/src/linkahead/common/models.py b/src/linkahead/common/models.py
index 6f38ff40a46b9bd9fe9236cc2cd4b071c2f2af7c..d8e6a367077b104d8b1ed5f3631dec470eaf8381 100644
--- a/src/linkahead/common/models.py
+++ b/src/linkahead/common/models.py
@@ -115,6 +115,7 @@ if TYPE_CHECKING:
                           "SUGGESTED", "RECOMMENDED", "ALL", "NONE"]
     IMPORTANCE = Literal["OBLIGATORY",
                          "RECOMMENDED", "SUGGESTED", "FIX", "NONE"]
+    ROLE = Literal["Entity", "Record", "RecordType", "Property", "File"]
 
 SPECIAL_ATTRIBUTES = ["name", "role", "datatype", "description",
                       "id", "path", "checksum", "size", "value"]
@@ -143,7 +144,7 @@ class Entity:
         **kwargs,
     ):
 
-        self.__role = kwargs["role"] if "role" in kwargs else None
+        self.__role: Optional[ROLE] = kwargs["role"] if "role" in kwargs else None
         self._checksum: Optional[str] = None
         self._size = None
         self._upload = None