diff --git a/docs/using_mlflow.md b/docs/using_mlflow.md
index 6773bb9..d44ad80 100644
--- a/docs/using_mlflow.md
+++ b/docs/using_mlflow.md
@@ -198,6 +198,9 @@ jaxtrain \
 ```bash
 mlflow ui
 # Opens http://localhost:5000
+
+# Sets up UI with tracking from .db
+mlflow server --backend-store-uri <path/to/tracking.db>
 ```
 
 
@@ -381,6 +384,9 @@ jaxtrain \
 # View experiments
 mlflow ui
 
+# Sets up UI with tracking from .db
+mlflow server --backend-store-uri <path/to/tracking.db>
+
 # Python queries
 python -c "
 import mlflow
@@ -423,6 +429,9 @@ jaxtrain \
 
 # 4. Review in UI
 mlflow ui
+
+# Sets up UI with tracking from .db
+mlflow server --backend-store-uri <path/to/tracking.db>
 ```
 
 ## 🔧 Troubleshooting
@@ -451,6 +460,15 @@ print(experiment.artifact_location)
 
 Ensure all files tracked in the data generation experiment exist in your training data folder. Check the error message for the list of missing files.
 
+### MLflow revision error
+```bash
+alembic.util.exc.CommandError: Can't locate revision identified by <revision number>
+```
+### Solution:
+```bash
+pip install --upgrade mlflow
+```
+
 ## 📚 See Also
 
 - [MLflow Documentation](https://mlflow.org/docs/latest/index.html)