diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index dcb62dc..b9a137e 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -25,7 +25,7 @@ docker_build:
       - ci/docker/**/*
 
 .test_base_task: &test_base_task
-  image: registry.gitlab.com/gift-surg/puma/puma-ci:master
+  image: registry.gitlab.com/gift-surg/puma/puma-ci:2-ensure-docker-build-runs
   script:
     - pytest