diff --git a/.github/workflows/nvidia-tests.yaml b/.github/workflows/nvidia-tests.yaml
index 1f8a7d5..4d83fd1 100644
--- a/.github/workflows/nvidia-tests.yaml
+++ b/.github/workflows/nvidia-tests.yaml
@@ -28,7 +28,7 @@ jobs:
         export G_data=$(( 2 / G_inter ))
         export memopt=${{ matrix.memopt }}
         echo "training with G_inter = ${G_inter}, G_data = $(( 2 / G_inter  )) ${{ matrix.memopt }}" 
-        mpirun -n 2 pytest --with-mpi ./tests/test_vit.py 
+        mpirun -n 2 pytest --with-mpi test_vit.py 
     - name: Uninstall AxoNN
       run: |
         pip uninstall --yes axonn
@@ -43,7 +43,7 @@ jobs:
         pip install -r requirements.txt
     - name: Run unit intra-layer unit tests 
       run: |
-        mpirun -n 2 pytest --with-mpi ./tests/test_intra_layer_fc.py 
+        mpirun -n 2 pytest --with-mpi test_intra_layer_fc.py 
     - name: Uninstall AxoNN
       run: |
         pip uninstall --yes axonn