diff --git a/CHANGELOG.md b/CHANGELOG.md
index 9bddb143c..a6ebcfffd 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -25,6 +25,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - Fixed timezone calculation in datapipe cosine zenith utility.
 - Refactored EDMPrecondSRV2 preconditioner and fixed the bug related to the metadata
 - Extended the checkpointing utility to store metadata.
+- Corrected missing export of loggin function used by transolver model
 
 ### Deprecated
 
diff --git a/modulus/launch/logging/__init__.py b/modulus/launch/logging/__init__.py
index fe4fa83db..50466a367 100644
--- a/modulus/launch/logging/__init__.py
+++ b/modulus/launch/logging/__init__.py
@@ -16,4 +16,5 @@
 
 from .console import PythonLogger, RankZeroLoggingWrapper
 from .launch import LaunchLogger
+from .mlflow import initialize_mlflow
 from .wandb import initialize_wandb