diff --git a/Jenkinsfile.train b/Jenkinsfile.train
index 22175b6..fb37e6b 100644
--- a/Jenkinsfile.train
+++ b/Jenkinsfile.train
@@ -1,6 +1,6 @@
 pipeline {
     agent {
-		docker { image 'kamilguttmann/ium:latest' }
+		docker { image 'kamilguttmann/ium:train' }
 	}
     parameters {
         string (