diff --git a/docs/conf.py b/docs/conf.py
index ba2dfdbb44..095152fac9 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -431,8 +431,6 @@ def delayed_setup(app: Sphinx) -> None:
 
 
 def setup(app: Sphinx) -> dict[str, bool]:
-    app.connect("builder-inited", delayed_setup, priority=0)  # type: ignore
-
+    app.connect("builder-inited", delayed_setup, priority=0)# type: ignore
     app.setup_extension("litestar_sphinx_theme")
-
     return {"parallel_read_safe": True, "parallel_write_safe": True}
diff --git a/tools/build_docs.py b/tools/build_docs.py
index 235b168b71..1f8a44c0e2 100644
--- a/tools/build_docs.py
+++ b/tools/build_docs.py
@@ -27,6 +27,7 @@
 
 parser = argparse.ArgumentParser()
 parser.add_argument("--version", required=False)
+parser.add_argument("--environment", required=False)
 parser.add_argument("output")
 
 
@@ -86,7 +87,7 @@ def build(output_dir: str, version: str | None, environment: str = "local") -> N
 
 def main() -> None:
     args = parser.parse_args()
-    build(output_dir=args.output, version=args.version)
+    build(output_dir=args.output, version=args.version, environment=args.environment)
 
 
 if __name__ == "__main__":