diff --git a/run_autoformat.sh b/run_autoformat.sh
index c5b9e24..90bf651 100755
--- a/run_autoformat.sh
+++ b/run_autoformat.sh
@@ -2,5 +2,6 @@
 yapf -i -r --style .style.yapf --exclude '**/third_party' llm4pddl
 yapf -i -r --style .style.yapf scripts
 yapf -i -r --style .style.yapf tests
+yapf -i -r --style .style.yapf *.py
 docformatter -i -r . --exclude venv llm4pddl/third_party
 isort .