diff --git a/app/services/gaze_tracker.py b/app/services/gaze_tracker.py
index 3354551..f3bd872 100644
--- a/app/services/gaze_tracker.py
+++ b/app/services/gaze_tracker.py
@@ -87,8 +87,7 @@ def trian_and_predict(model_name, X_train, y_train, X_test, y_test, label):
     """
     if (
         model_name == "Linear Regression"
-        or model_name == "Elastic Net"
-        or model_name == "Support Vector Regressor"
+        or model_name == "Random Forest Regressor"
     ):
         model = models[model_name]
         model.fit(X_train, y_train)