diff --git a/template/python3-flask-debian/Dockerfile b/template/python3-flask-debian/Dockerfile
index f0a245d..6086cf9 100644
--- a/template/python3-flask-debian/Dockerfile
+++ b/template/python3-flask-debian/Dockerfile
@@ -48,7 +48,7 @@ ARG TEST_ENABLED=true
 RUN [ "$TEST_ENABLED" = "false" ] && echo "skipping tests" || eval "$TEST_COMMAND"
 
 
-FROM build as ship
+FROM test as ship
 WORKDIR /home/app/
 
 #configure WSGI server and healthcheck
diff --git a/template/python3-flask/Dockerfile b/template/python3-flask/Dockerfile
index 8235eba..748fbad 100644
--- a/template/python3-flask/Dockerfile
+++ b/template/python3-flask/Dockerfile
@@ -46,7 +46,7 @@ ARG TEST_COMMAND=tox
 ARG TEST_ENABLED=true
 RUN [ "$TEST_ENABLED" = "false" ] && echo "skipping tests" || eval "$TEST_COMMAND"
 
-FROM build as ship
+FROM test as ship
 WORKDIR /home/app/
 
 #configure WSGI server and healthcheck
diff --git a/template/python3-http-debian/Dockerfile b/template/python3-http-debian/Dockerfile
index 99a6c42..db9fadf 100644
--- a/template/python3-http-debian/Dockerfile
+++ b/template/python3-http-debian/Dockerfile
@@ -44,7 +44,7 @@ ARG TEST_ENABLED=true
 RUN [ "$TEST_ENABLED" = "false" ] && echo "skipping tests" || eval "$TEST_COMMAND"
 
 
-FROM build as ship
+FROM test as ship
 WORKDIR /home/app/
 
 USER app
diff --git a/template/python3-http/Dockerfile b/template/python3-http/Dockerfile
index 626d46c..8448c18 100644
--- a/template/python3-http/Dockerfile
+++ b/template/python3-http/Dockerfile
@@ -43,7 +43,7 @@ ARG TEST_COMMAND=tox
 ARG TEST_ENABLED=true
 RUN [ "$TEST_ENABLED" = "false" ] && echo "skipping tests" || eval "$TEST_COMMAND"
 
-FROM build as ship
+FROM test as ship
 WORKDIR /home/app/
 
 # configure WSGI server and healthcheck