diff --git a/src/linkahead/utils/get_entity.py b/src/linkahead/utils/get_entity.py
index 8d280521364a2597419577ef58897aa2d98ccdaa..f84dc107e275390e53c6127834f53e5e5c6521cd 100644
--- a/src/linkahead/utils/get_entity.py
+++ b/src/linkahead/utils/get_entity.py
@@ -21,7 +21,7 @@
 
 """Convenience functions to retrieve a specific entity."""
 
-from typing import Union
+from typing import Union, Optional
 
 from ..common.models import Entity, execute_query
 from .escape import escape_squoted_text