diff --git a/train_sa.py b/train_sa.py
index 650d88eb6d504fbc870226683cfdb0be3203c275..5d763a05fb3727e1af42ceaf5a891ea374cfca73 100644
--- a/train_sa.py
+++ b/train_sa.py
@@ -81,7 +81,7 @@ def main():
 
     trainer = pl.Trainer(accelerator="gpu", 
                          devices=num_gpus, 
-                         profiler="simple" if args.proffiler else None, 
+                         profiler="simple" if args.profiler else None, 
                          default_root_dir="./logs", 
                          logger=WandbLogger(project="slot-att", offline=True) if args.wandb else None,
                          strategy="ddp_find_unused_parameters_true" if num_gpus > 1 else "auto",