diff --git a/tools/program.py b/tools/program.py
index 511ee9dd1f12273eb773b6f2e29a3955940721ee..b01c2e43fe2239862205d16bb6c523b8aa45cca7 100755
--- a/tools/program.py
+++ b/tools/program.py
@@ -697,7 +697,7 @@ def preprocess(is_train=False):
             wandb_params = config['wandb']
         else:
             wandb_params = dict()
-        wandb_params.update({'save_dir': save_model_dir})
+        wandb_params.update({'save_dir': save_dir})
         log_writer = WandbLogger(**wandb_params, config=config)
         loggers.append(log_writer)
     else: