diff --git a/cleanup.sh b/cleanup.sh
index aa6deeb..4cfc7a8 100755
--- a/cleanup.sh
+++ b/cleanup.sh
@@ -1,3 +1,3 @@
-autoflake -i --remove-all-unused-imports src/wireviz/*.py
-isort src/wireviz/*py
-black src/wireviz/*.py
+autoflake -i -r --remove-all-unused-imports src/wireviz/
+isort src/wireviz/*.py src/wireviz/tools/*.py
+black src/wireviz/*.py src/wireviz/tools/*.py