diff --git a/cognee/modules/storage/utils/__init__.py b/cognee/modules/storage/utils/__init__.py
index 7073e6470..a399e8a82 100644
--- a/cognee/modules/storage/utils/__init__.py
+++ b/cognee/modules/storage/utils/__init__.py
@@ -29,7 +29,9 @@ def copy_model(model: DataPoint, include_fields: dict = {}, exclude_fields: list
         **include_fields
     }
 
-    return create_model(model.__name__, **final_fields)
+    model = create_model(model.__name__, **final_fields)
+    model.model_rebuild()
+    return model
 
 def get_own_properties(data_point: DataPoint):
     properties = {}