diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 03c012de93d1f31241336a0def47dc4524bcd801..e6084b1dbd0ea87753459ae0a0ab0bdc539e0e37 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -18,6 +18,9 @@ before_script:
 
 train:
   stage: train
+   tags:
+    - pages
+
   script:
     - source venv/bin/activate
     - python train_model.py
@@ -26,14 +29,11 @@ train:
       - trained_model.joblib
       - tokenizer.joblib
 
-test:
-  stage: test
-  script:
-    - source venv/bin/activate
-    - python test_model.py
-
 deploy:
   stage: deploy
+   tags:
+    - pages
+
   script:
     - echo "Deploying model locally..."
     - source venv/bin/activate