diff --git a/Jenkinsfile.train b/Jenkinsfile.train
index 420d805..2b20986 100644
--- a/Jenkinsfile.train
+++ b/Jenkinsfile.train
@@ -1,6 +1,6 @@
 pipeline {
     agent {
-		dockerfile true
+		docker { image 'kamilguttmann/ium:trainv2' }
 	}
     parameters {
         string (