diff --git a/adapta/ml/mlflow/_client.py b/adapta/ml/mlflow/_client.py
index 5648f75d..2e1c1acf 100644
--- a/adapta/ml/mlflow/_client.py
+++ b/adapta/ml/mlflow/_client.py
@@ -166,7 +166,8 @@ def create_run(self, experiment_name: str, run_name: str) -> str:
         :return: run id
         """
         experiment = self._client.get_experiment_by_name(experiment_name)
-        return self._client.create_run(experiment_id=experiment.experiment_id, run_name=run_name).info.run_id
+        experiment_id = experiment.experiment_id if experiment else self._client.create_experiment(experiment_name)
+        return self._client.create_run(experiment_id=experiment_id, run_name=run_name).info.run_id
 
     def terminate_run(self, run_id: str):
         """