diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 17221c8..c48e492 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -25,6 +25,7 @@ stages: # List of stages for jobs, and their order of execution
 .pip-test-base: # This job runs in the test stage.
   stage: test    # It only starts when the job in the build stage completes successfully.
   script:
+    - pip install wheel numpy gpytorch==1.8.1
     - pip install .
     - pip install pytest pytest-cov coverage
     - coverage --version
@@ -149,4 +150,4 @@ deploy-public-repo:
     - git config user.email "${EMAIL}"
     - git config user.name "${NAME}"
     - git commit -m "mirror to GitHub"
-    - git push --force https://${GH_USERNAME}:${GH_ACCESS_KEY}@github.com/${GH_USERNAME}/BAxUS.git orphan:main
\ No newline at end of file
+    - git push --force https://${GH_USERNAME}:${GH_ACCESS_KEY}@github.com/${GH_USERNAME}/BAxUS.git orphan:main