diff --git a/{{cookiecutter.project_slug}}/.gitlab-ci.yml b/{{cookiecutter.project_slug}}/.gitlab-ci.yml
index 60925b81..fcdfb7dd 100644
--- a/{{cookiecutter.project_slug}}/.gitlab-ci.yml
+++ b/{{cookiecutter.project_slug}}/.gitlab-ci.yml
@@ -21,7 +21,6 @@ flake8:
 
 pytest:
   stage: test
-  image: python:3.8
   {% if cookiecutter.use_docker == 'y' -%}
   image: docker/compose:latest
   tags:
@@ -36,6 +35,7 @@ pytest:
   script:
     - docker-compose -f local.yml run django pytest
   {%- else %}
+  image: python:3.8
   tags:
     - python
   services: