diff --git a/.vscode/tasks.json b/.vscode/tasks.json
index f0e2790..c3f6d4d 100644
--- a/.vscode/tasks.json
+++ b/.vscode/tasks.json
@@ -12,13 +12,13 @@
         {
             "label": "Test (prod1)",
             "type": "shell",
-            "command": "./test/test.sh prod 1",
+            "command": "./test/test.sh prod1 1",
             "group": "build"
         },
         {
             "label": "Test (prod2)",
             "type": "shell",
-            "command": "./test/test.sh prod 1",
+            "command": "./test/test.sh prod2 1",
             "group": "build"
         },
         {
diff --git a/docs/bf2hub-bf2stats-example/docker-compose.build.yml b/docs/bf2hub-bf2stats-example/docker-compose.build.yml
index 9b6e570..02891ce 100644
--- a/docs/bf2hub-bf2stats-example/docker-compose.build.yml
+++ b/docs/bf2hub-bf2stats-example/docker-compose.build.yml
@@ -1,5 +1,10 @@
 # This is a docker compose override file, for production builds with caching for CI environments
 services:
+  bf2:
+    image: startersclan/docker-bf2:v1.5.3153.0
+    volumes:
+      - ../../src/python:/server/bf2/python # Mount the python files
+
   asp:
     build:
       dockerfile: Dockerfile.asp
diff --git a/docs/full-bf2-stack-example/docker-compose.build.yml b/docs/full-bf2-stack-example/docker-compose.build.yml
index 9b6e570..02891ce 100644
--- a/docs/full-bf2-stack-example/docker-compose.build.yml
+++ b/docs/full-bf2-stack-example/docker-compose.build.yml
@@ -1,5 +1,10 @@
 # This is a docker compose override file, for production builds with caching for CI environments
 services:
+  bf2:
+    image: startersclan/docker-bf2:v1.5.3153.0
+    volumes:
+      - ../../src/python:/server/bf2/python # Mount the python files
+
   asp:
     build:
       dockerfile: Dockerfile.asp