diff --git a/.gitlab/ci/ubuntu_python.gitlab-ci.yml b/.gitlab/ci/ubuntu_python.gitlab-ci.yml
index a541e34b95053020c3d573c4ae11e5a94d1b533e..5cdd941e019da4c75d95aeb71db60ee796a0e173 100644
--- a/.gitlab/ci/ubuntu_python.gitlab-ci.yml
+++ b/.gitlab/ci/ubuntu_python.gitlab-ci.yml
@@ -76,7 +76,11 @@ coverage:ubuntu_python:
     - source venv/bin/activate
     - which python
     - python -m pip list
-    - python -m pip install numpy coverage requests unittest-xml-reporting pytest onnxruntime
+    - python -m pip install numpy coverage requests unittest-xml-reporting 
+    - >
+      if [[ "$CI_PROJECT_NAME" == "aidge_onnx" ]]; then
+        python -m pip install pytest onnxruntime>=1.18.0
+      fi
   script:
     - cd ${CI_PROJECT_NAME}
     # Retrieve the installation path of the module, since it is installed with pip.