diff --git a/test/pytest/ci-template.yml b/test/pytest/ci-template.yml
index afaf90da4d..f6aa700415 100644
--- a/test/pytest/ci-template.yml
+++ b/test/pytest/ci-template.yml
@@ -1,6 +1,6 @@
 .pytest:
   stage: test
-  image: gitlab-registry.cern.ch/fastmachinelearning/hls4ml-testing:0.5.5.base
+  image: gitlab-registry.cern.ch/fastmachinelearning/hls4ml-testing:0.5.6.base
   tags:
     - k8s-default
   before_script: