diff --git a/mlflow_prediction_registry.py b/mlflow_prediction_registry.py
index f8c5736..4507887 100644
--- a/mlflow_prediction_registry.py
+++ b/mlflow_prediction_registry.py
@@ -22,7 +22,7 @@ experiment = client.get_latest_versions(model_name, stages=None)
 
 print(experiment)
 
-with open(f'{experiment.source}/{input}', 'r') as file:
+with open(f'{experiment[0].source}/{input}', 'r') as file:
     json_data = json.load(file)
 
 print(model(torch.tensor(np.array(json_data['inputs'])).float()))
\ No newline at end of file