diff --git a/imperative/python/test/run.sh b/imperative/python/test/run.sh
index 0f1f0fd7..967740a8 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