diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 1880bd386f54e9310f91f90f7c626087f95ea95b..f31eb283f4e74f1bfb4a6a9533b8cc1c03657b40 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -76,6 +76,7 @@ flake8-lint:
   except:
     variables:
       - $ENABLE_NIGHTLY_BUILDS
+  image: machine_learning.Dockerfile
   script:
     - flake8 src/pystencils_rec
   tags: