diff --git a/docker-compose-prod.yml b/docker-compose-prod.yml
index 0d4fa846..011de723 100644
--- a/docker-compose-prod.yml
+++ b/docker-compose-prod.yml
@@ -7,7 +7,7 @@ services:
     container_name: medcattrainer
     hostname: medcat
     restart: always
-    image: cogstacksystems/medcat-trainer:v2.5.3
+    image: cogstacksystems/medcat-trainer:v2.5.4
     volumes:
       - ./configs:/home/configs
       - api-media:/home/api/media
@@ -18,12 +18,12 @@ services:
     env_file:
       - ./envs/env-prod
     environment:
-      - MCT_VERSION=v2.5.3
+      - MCT_VERSION=v2.5.4
     command: /home/run.sh
 
   nginx:
     container_name: medcattrainer_nginx
-    image: cogstacksystems/medcat-trainer-nginx:v2.5.3
+    image: cogstacksystems/medcat-trainer-nginx:v2.5.4
     restart: always
     volumes:
       - api-media:/home/api/media
diff --git a/docker-compose.yml b/docker-compose.yml
index af1d46e3..787df6e0 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -5,7 +5,7 @@ version: '3.4'
 
 services:
   medcattrainer:
-    image: cogstacksystems/medcat-trainer:v2.5.3
+    image: cogstacksystems/medcat-trainer:v2.5.4
     restart: always
     volumes:
       - ./configs:/home/configs
@@ -15,11 +15,11 @@ services:
     env_file:
       - ./envs/env
     environment:
-      - MCT_VERSION=v2.5.3
+      - MCT_VERSION=v2.5.4
     command: /home/run.sh
 
   nginx:
-    image: cogstacksystems/medcat-trainer-nginx:v2.5.3
+    image: cogstacksystems/medcat-trainer-nginx:v2.5.4
     restart: always
     volumes:
       - api-media:/home/api/media