diff --git a/tests/build.yml b/tests/build.yml
index a2b4739c..0d89c5b8 100644
--- a/tests/build.yml
+++ b/tests/build.yml
@@ -52,7 +52,10 @@ services:
 
   docs:
     image: ${DOCKER_ORG:-mailu}/${DOCKER_PREFIX}docs:${MAILU_VERSION:-local}
-    build: ../docs
+    build:
+      context: ../docs
+      args:
+        version: ${MAILU_VERSION:-local}
 
   setup:
     image: ${DOCKER_ORG:-mailu}/${DOCKER_PREFIX}setup:${MAILU_VERSION:-local}