diff --git a/01_causal_discovery.ipynb b/01_causal_discovery.ipynb
index ec47945..d26643c 100644
--- a/01_causal_discovery.ipynb
+++ b/01_causal_discovery.ipynb
@@ -491,7 +491,7 @@
    "language": "python",
    "notebookMetadata": {
     "mostRecentlyExecutedCommandWithImplicitDF": {
-     "commandId": 888983575928260,
+     "commandId": 2548317818328688,
      "dataframes": [
       "_sqldf"
      ]
diff --git a/util/notebook-config.py b/util/notebook-config.py
index cb8f2a3..0519068 100644
--- a/util/notebook-config.py
+++ b/util/notebook-config.py
@@ -17,6 +17,8 @@
 %matplotlib inline
 %config InlineBackend.figure_format = "retina"
 
+mlflow.set_registry_uri("databricks-uc")
+
 # COMMAND ----------
 
 # Create catalog if it doesn't exist