diff --git a/train.py b/train.py
index 26765faf..e31713c6 100644
--- a/train.py
+++ b/train.py
@@ -44,6 +44,7 @@ def eval_metrics(actual, pred):
     alpha = float(sys.argv[1]) if len(sys.argv) > 1 else 0.5
     l1_ratio = float(sys.argv[2]) if len(sys.argv) > 2 else 0.5
 
+    mlflow.set_experiment("/wine-quality-experiment")
     with mlflow.start_run():
         lr = ElasticNet(alpha=alpha, l1_ratio=l1_ratio, random_state=42)
         lr.fit(train_x, train_y)