diff --git a/build.sh b/build.sh
index c6906357..39192c78 100755
--- a/build.sh
+++ b/build.sh
@@ -130,11 +130,11 @@ case "$1" in
    ;;
 
    test)
-      python3 -m unittest discover test 'test_*.py' 
+      python3 -m unittest discover lib/cdist/test 'test_*.py' 
    ;;
 
    test-all)
-      python3 -m unittest discover test '*.py' 
+      python3 -m unittest discover lib/cdist/test '*.py' 
    ;;
 
    *)