diff --git a/wtpsplit/train/train_adapter.py b/wtpsplit/train/train_adapter.py
index e4435290..e7e7a031 100644
--- a/wtpsplit/train/train_adapter.py
+++ b/wtpsplit/train/train_adapter.py
@@ -374,7 +374,8 @@ def maybe_pad(text):
 
     # 1 wandb run for all language-dataset combinations
     if "wandb" in training_args.report_to and training_args.process_index == 0:
-        wandb.init(name=wandb_name, project=args.wandb_project, group=wandb_name)
+        # TODO: don't hardcode entity
+        wandb.init(name=wandb_name, project=args.wandb_project, group=wandb_name, entity="markus_583")
         wandb.config.update(args)
         wandb.config.update(training_args)
         wandb.config.update(label_args)