diff --git a/cognee/shared/GithubTopology.py b/cognee/shared/GithubTopology.py
index a84aa5983..3315c72e4 100644
--- a/cognee/shared/GithubTopology.py
+++ b/cognee/shared/GithubTopology.py
@@ -18,7 +18,7 @@ class Directory(BaseModel):
     directories: List['Directory'] = []
 
 # Allows recursive Directory Model
-Directory.update_forward_refs()
+Directory.model_rebuild()
 
 class RepositoryProperties(BaseModel):
     custom_properties: Optional[Dict[str, Any]] = None
diff --git a/cognee/tasks/infer_data_ontology/models/models.py b/cognee/tasks/infer_data_ontology/models/models.py
index b62bf3ac0..9c086b5c7 100644
--- a/cognee/tasks/infer_data_ontology/models/models.py
+++ b/cognee/tasks/infer_data_ontology/models/models.py
@@ -12,7 +12,7 @@ class NodeModel(BaseModel):
     default_relationship: Optional[RelationshipModel] = None
     children: List[Union[Dict[str, Any], "NodeModel"]] = Field(default_factory=list)
 
-NodeModel.update_forward_refs()
+NodeModel.model_rebuild()
 
 
 class OntologyNode(BaseModel):