diff --git a/.just/documentation.just b/.just/documentation.just
index 9ce17f1..5e1b1fc 100644
--- a/.just/documentation.just
+++ b/.just/documentation.just
@@ -12,15 +12,17 @@ fmt:
 
 # Build documentation using Sphinx
 [no-cd]
-build LOCATION="docs/_build/html": cog
+build LOCATION="docs/_build/html":
+    @just cog
     uv run --extra docs sphinx-build docs {{ LOCATION }}
 
 # Serve documentation locally
 [no-cd]
-serve PORT="8000": cog
+serve PORT="8000":
     #!/usr/bin/env sh
     HOST="localhost"
     if [ -f "/.dockerenv" ]; then
         HOST="0.0.0.0"
     fi
+    just cog
     uv run --extra docs sphinx-autobuild docs docs/_build/html --host "$HOST" --port {{ PORT }}