diff --git a/google/cloud/aiplatform/tensorboard/uploader_tracker.py b/google/cloud/aiplatform/tensorboard/uploader_tracker.py
index 0fc4c1c042..daf0f6541d 100644
--- a/google/cloud/aiplatform/tensorboard/uploader_tracker.py
+++ b/google/cloud/aiplatform/tensorboard/uploader_tracker.py
@@ -254,7 +254,7 @@ def _create_uploader(
             allowed_plugins=uploader_constants.ALLOWED_PLUGINS,
             writer_client=api_client,
             logdir=logdir,
-            one_shot=True,
+            one_shot=one_shot,
             run_name_prefix=run_name_prefix,
             description=description,
             verbosity=verbosity,