diff --git a/e2e-test/generator-alfresco-docker-compose/generators/app/templates/6.1/docker-compose-ee.yml b/e2e-test/generator-alfresco-docker-compose/generators/app/templates/6.1/docker-compose-ee.yml
index 0d3fce47a..88a02e9e6 100755
--- a/e2e-test/generator-alfresco-docker-compose/generators/app/templates/6.1/docker-compose-ee.yml
+++ b/e2e-test/generator-alfresco-docker-compose/generators/app/templates/6.1/docker-compose-ee.yml
@@ -151,9 +151,9 @@ services:
         build:
           context: ./zeppelin
           args:
-            - ZEPPELIN_TAG=${ZEPPELIN_TAG}
-        environment:
+            ZEPPELIN_TAG: ${ZEPPELIN_TAG}
             ALFRESCO_COMMS: <%=secureComms%>
+        environment:
             REPO_HOST: "alfresco"
             REPO_PORT: "<%=alfrescoPort%>" <% if (httpMode == 'https') { %>
             JAVA_OPTS: "