diff --git a/.gitlab/ci/ubuntu_python.gitlab-ci.yml b/.gitlab/ci/ubuntu_python.gitlab-ci.yml
index 5cdd941e019da4c75d95aeb71db60ee796a0e173..f52c94faa33cde5a79386765e7cbfed595dd0aeb 100644
--- a/.gitlab/ci/ubuntu_python.gitlab-ci.yml
+++ b/.gitlab/ci/ubuntu_python.gitlab-ci.yml
@@ -13,7 +13,7 @@ build:ubuntu_python:
   before_script:
     - !reference [.retrieve_deps:apt, script]
     - DEPENDENCY_JOB="build:ubuntu_python"
-    - !reference [.ubuntu:download:artifacts, before_script] 
+    - !reference [.ubuntu:download:artifacts, before_script]
     - python3 -m pip install virtualenv
 
   script:
@@ -44,7 +44,8 @@ test:ubuntu_python:
   before_script:
     - !reference [.retrieve_deps:apt, script]
     - source venv/bin/activate
-    - python -m pip install unittest-xml-reporting pytest
+    # - python -m pip install unittest-xml-reporting pytest
+    - python -m pip install pytest
     - which python
     - python -m pip list
     - >
@@ -53,9 +54,10 @@ test:ubuntu_python:
       fi
   script:
     - cd ${CI_PROJECT_NAME}
-    # Run on discovery all tests located in project/unit_tests/python and discard the stdout 
+    # Run on discovery all tests located in project/unit_tests/python and discard the stdout
     # only to show the errors/warnings and the results of the tests
-    - python -m xmlrunner discover -s unit_tests/ -v -b --output-file xmlrunner-results.xml
+    # - python -m xmlrunner discover -s unit_tests/ -v -b --output-file xmlrunner-results.xml
+    - pytest unit_tests/ -v --junitxml=xmlrunner-results.xml
 
   artifacts:
     reports:
@@ -71,12 +73,12 @@ coverage:ubuntu_python:
   tags:
     - docker
 
-  before_script: 
+  before_script:
     - !reference [.retrieve_deps:apt, script]
     - source venv/bin/activate
     - which python
     - python -m pip list
-    - python -m pip install numpy coverage requests unittest-xml-reporting 
+    - 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