diff --git a/mlflow_prediction_registry.py b/mlflow_prediction_registry.py
index 4753dc4..25b700e 100644
--- a/mlflow_prediction_registry.py
+++ b/mlflow_prediction_registry.py
@@ -13,7 +13,7 @@ arguments = sys.argv[1:]
 
 mlflow.set_tracking_uri("http://172.17.0.1:5000")
 client = MlflowClient()
-version = 1
+model_version = 1
 model_name = "s426206"
 
 input = str(arguments[0])