diff --git a/src/linkahead/high_level_api.py b/src/linkahead/high_level_api.py
index 49cb51d8d3ab828e38663292a11ced87380c4d7d..96fc09af3b59ef4a1061908ec1ac241a256acaa4 100644
--- a/src/linkahead/high_level_api.py
+++ b/src/linkahead/high_level_api.py
@@ -264,7 +264,7 @@ class LinkAheadPythonEntity(object):
 
     def _set_property_from_entity(self, ent: db.Entity, importance: str,
                                   references: Optional[db.Container],
-                                  visited: Dict[int, "CaosDBPythonEntity"]):
+                                  visited: Dict[int, "LinkAheadPythonEntity"]):
         """
         Set a new property using an entity from the normal python API.
 
@@ -382,7 +382,7 @@ class LinkAheadPythonEntity(object):
                              val: List,
                              pr: str,
                              references: Optional[db.Container],
-                             visited: Dict[int, "CaosDBPythonEntity"]):
+                             visited: Dict[int, "LinkAheadPythonEntity"]):
         """
         Convert a list to a python list of the correct type.
 
@@ -403,7 +403,7 @@ class LinkAheadPythonEntity(object):
                               val: Any,
                               pr: str,
                               references: Optional[db.Container],
-                              visited: Dict[int, "CaosDBPythonEntity"]):
+                              visited: Dict[int, "LinkAheadPythonEntity"]):
         """
         Convert val to the correct type which is indicated by the database
         type string in pr.
@@ -810,7 +810,7 @@ def _single_convert_to_python_object(robj: LinkAheadPythonEntity,
                                      entity: db.Entity,
                                      references: Optional[db.Container] = None,
                                      visited: Optional[Dict[int,
-                                                            "CaosDBPythonEntity"]] = None):
+                                                            "LinkAheadPythonEntity"]] = None):
     """
     Convert a db.Entity from the standard API to a (previously created)
     LinkAheadPythonEntity from the high level API.
@@ -941,7 +941,7 @@ def convert_to_entity(python_object):
 def convert_to_python_object(entity: Union[db.Container, db.Entity],
                              references: Optional[db.Container] = None,
                              visited: Optional[Dict[int,
-                                                    "CaosDBPythonEntity"]] = None):
+                                                    "LinkAheadPythonEntity"]] = None):
     """
     Convert either a container of LinkAhead entities or a single LinkAhead entity
     into the high level representation.