diff --git a/integration/monai/examples/mednist/code/monai_mednist_train.py b/integration/monai/examples/mednist/code/monai_mednist_train.py
index bb8dfb41ea..45033c2aa2 100644
--- a/integration/monai/examples/mednist/code/monai_mednist_train.py
+++ b/integration/monai/examples/mednist/code/monai_mednist_train.py
@@ -52,9 +52,6 @@
 
 print_config()
 
-# (2) initializes NVFlare client API
-flare.init()
-
 # Setup data directory
 directory = os.environ.get("MONAI_DATA_DIRECTORY")
 root_dir = tempfile.mkdtemp() if directory is None else directory
@@ -97,6 +94,9 @@
     train_handlers=StatsHandler(),
 )
 
+# (2) initializes NVFlare client API
+flare.init()
+
 # (optional) calculate total steps
 steps = max_epochs * len(train_loader)
 # Run the training