diff --git a/Jenkinsfile b/Jenkinsfile
index 68812e18..60639043 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -26,7 +26,7 @@ pipeline {
                         }
                     }
                     steps {
-                        sh 'pytest ./tests/core/drivers/torch_driver --durations=0 -m torch'
+                        sh 'pytest ./tests/ --durations=0 -m torch'
                     }
                 }
                 stage('Test Torch-1.6') {
@@ -37,7 +37,7 @@ pipeline {
                         }
                     }
                     steps {
-                        sh 'pytest ./tests/core/drivers/torch_driver --durations=0 -m torch'
+                        sh 'pytest ./tests/ --durations=0 -m torch'
                     }
                 }
                 stage('Test Paddle') {