diff --git a/format_code.sh b/format_code.sh
index aff712cb..e3320c5b 100755
--- a/format_code.sh
+++ b/format_code.sh
@@ -1 +1 @@
-docker run -ti -v $(pwd):/tmp/python python bash -c "cd /tmp/python ; pip install black; black ."
+docker run --rm -ti -v $(pwd):/tmp/python/app python bash -c "cd /tmp/python ; pip install black; black ."