1 files changed, 8 insertions, 3 deletions
diff --git a/backend/api/api/Services/ModelService.cs b/backend/api/api/Services/ModelService.cs
index 33dea30e..a3939b29 100644
--- a/backend/api/api/Services/ModelService.cs
+++ b/backend/api/api/Services/ModelService.cs
@@ -13,7 +13,7 @@ namespace api.Services
         public ModelService(IUserStoreDatabaseSettings settings, IMongoClient mongoClient)
         {
             var database = mongoClient.GetDatabase(settings.DatabaseName);
-            _model = database.GetCollection<Model>(settings.ModelCollectionName);
+            _model = database.GetCollection<Model>(settings.PredictorCollectionName);
         }
 
         public Model Create(Model model)
@@ -27,11 +27,16 @@ namespace api.Services
             _model.DeleteOne(model => (model.username == username && model.name == name));
         }
 
-        public List<Model> GetAllModels(string username)
+        public List<Model> GetMyModels(string username)
         {
             return _model.Find(model => model.username == username).ToList();
         }
-
+        /*
+        public List<Model> GetPublicModels()
+        {
+            return _model.Find(model => model.isPublic == true).ToList();
+        }
+        */
         public Model GetOneModel(string username, string name)
         {
             return _model.Find(model => model.username == username && model.name == name).FirstOrDefault();