diff --git a/tools/program.py b/tools/program.py
index a8373435c..209212fde 100755
--- a/tools/program.py
+++ b/tools/program.py
@@ -685,7 +685,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: