diff --git a/train-mlflow.py b/train-mlflow.py
index 8a59205..1d56809 100644
--- a/train-mlflow.py
+++ b/train-mlflow.py
@@ -106,7 +106,7 @@ def my_main(epochs):
 
         
 
-            input_example = X_testing[0]
+            input_example = np.array(X_testing[0])
             siganture = mlflow.models.signature.infer_signature(np.array(X_training), np.array(y_predicted_train))
             tracking_url_type_store = urlparse(mlflow.get_tracking_uri()).scheme