diff --git a/Jenkinsfile_train b/Jenkinsfile_train
index e054272..1761e37 100644
--- a/Jenkinsfile_train
+++ b/Jenkinsfile_train
@@ -1,5 +1,5 @@
 pipeline {
-    agent {docker { image 'adnovac/ium_s434760:1.2' }}
+    agent {dockerfile true}
 	parameters{
         buildSelector(
             defaultSelector: lastSuccessful(),