diff --git a/Jenkinsfile_mlflow b/Jenkinsfile_mlflow
index aa0b4d6..4366f19 100644
--- a/Jenkinsfile_mlflow
+++ b/Jenkinsfile_mlflow
@@ -1,7 +1,9 @@
 pipeline {
-    docker {
+    agent {
+        docker {
             image 'ksero/ium:mlflow'
         }
+    } 
     stages {
         stage('Prediction') {
             steps {