diff --git a/train-model.Jenkinsfile b/train-model.Jenkinsfile
index b51c215..d3ccc0d 100644
--- a/train-model.Jenkinsfile
+++ b/train-model.Jenkinsfile
@@ -10,7 +10,7 @@ pipeline {
     stages {
         stage('Build') {
             steps {
-                cleanWs()
+//                 cleanWs()
                 git 'https://git.wmi.amu.edu.pl/s434749/ium_434749.git'
                 copyArtifacts fingerprintArtifacts: true, projectName: 's434749-create-dataset', selector: lastSuccessful()
                 sh "python3 train_model.py with 'batch_size=${params.batch_size}' 'learning_rate=${params.learning_rate}' 'epochs=${params.epochs}'"
diff --git a/train_model.py b/train_model.py
index 76ac7ed..8fe1dff 100644
--- a/train_model.py
+++ b/train_model.py
@@ -193,10 +193,10 @@ def cfg():
 
 
 def signature(model,in_alphabet,max_len):
-    mock_x = [('abc', 'xyz'), ('hey', 'man')]
+    mock_x = [('abc', 'xyz')]
     mock_text, _ = encode_str(mock_x, in_alphabet, max_len)
     mock_y = model(mock_text)
-    return mlflow.models.signature.infer_signature(mock_x, mock_y)
+    return mlflow.models.signature.infer_signature(mock_text, mock_y)
 
 @ex.automain
 def run(kernel_size, hidden_layers, data_file, epochs, teacher_forcing_probability, learning_rate, batch_size, max_len,