diff --git a/test/test-python-proto-consistency.sh b/test/test-python-proto-consistency.sh
index b938b01..27e6f15 100755
--- a/test/test-python-proto-consistency.sh
+++ b/test/test-python-proto-consistency.sh
@@ -33,7 +33,7 @@
 fi
 
 # delete and regenerate python protos
-rm -rf python/dmi.pb2 python/*/*.pb2.py python_temp
+rm -rf python/dmi/*
 make python-protos
 
 # Running git status ensures correct git diff-index picks up changes