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