diff --git a/.github/workflows/deploy.yaml b/.github/workflows/deploy.yaml
index eb5e804..cbbc7d2 100644
--- a/.github/workflows/deploy.yaml
+++ b/.github/workflows/deploy.yaml
@@ -8,7 +8,6 @@ on:
   push:
     branches:
       - main
-      - bugfix/docker
 
   # Allows you to run this workflow manually from the Actions tab
   workflow_dispatch:
diff --git a/tsadar_app.py b/tsadar_app.py
index f94f46a..ef9d429 100644
--- a/tsadar_app.py
+++ b/tsadar_app.py
@@ -3,7 +3,7 @@
 config.update("jax_enable_x64", True)
 
 import streamlit as st
-import yaml, os
+import yaml, os, mlflow
 from flatten_dict import flatten, unflatten
 
 from tsadar_gui import config
@@ -81,6 +81,7 @@ def process_file(file_path):
         with c2:
             if st.button("Fit"):
                 # run and wait for the results
+                mlflow.set_experiment(cfg["mlflow"]["experiment"])
                 run_id = run_for_app(cfg, mode="fit")
 
                 st.write(f"The results can be found at the mlflow run id {run_id}")