diff --git a/mainscripts/Trainer.py b/mainscripts/Trainer.py
index e212d2c..fa1069e 100644
--- a/mainscripts/Trainer.py
+++ b/mainscripts/Trainer.py
@@ -603,7 +603,7 @@ def main(**kwargs):
                         selected_preview = selected_preview % len(previews)
                         update_preview = True
                 elif op == 'tb':
-                    handle_tensorboard_op(input)
+                    handle_tensorboard_op(item)
                 elif op == 'close':
                     break