diff --git a/predictMlflow.py b/predictMlflow.py
index aa2061b..cfdefb7 100644
--- a/predictMlflow.py
+++ b/predictMlflow.py
@@ -2,7 +2,7 @@ import mlflow
 import numpy as np
 import json
 
-logged_model = '/mlruns/20/80fe21a0804844088147d15a3cebb3e5/artifacts/lego'
+logged_model = '/mlruns/14/80fe21a0804844088147d15a3cebb3e5/artifacts/lego-model'
 loaded_model = mlflow.pyfunc.load_model(logged_model)
 
 with open(f'{logged_model}/input_example.json') as f: