diff --git a/embedding_studio/data_access/mongo/mongo_dao.py b/embedding_studio/data_access/mongo/mongo_dao.py
index 866b2e8..4f706f7 100644
--- a/embedding_studio/data_access/mongo/mongo_dao.py
+++ b/embedding_studio/data_access/mongo/mongo_dao.py
@@ -124,7 +124,7 @@ def get_schema_properties(self) -> Set[str]:
 
         :return: Set of property names
         """
-        schema = self.model.schema()
+        schema = self.model.model_json_schema()
         return set(schema["properties"].keys())
 
     def get_model_projection(self) -> Dict[str, bool]: