diff --git a/imperative/python/test/run.sh b/imperative/python/test/run.sh
index 0f1f0fd79d491e33efbc620924cd2cb8f50a7c23..967740a87fe4c7ebffaaabdffb2dbd0fdc328879 100755
--- a/imperative/python/test/run.sh
+++ b/imperative/python/test/run.sh
@@ -1,6 +1,6 @@
 #!/bin/bash -e
 
-test_dirs="test"
+test_dirs="test megengine"
 TEST_PLAT=$1
 
 if [[ "$TEST_PLAT" == cpu ]]; then