diff --git a/slurm/run_dppo_dev.sh b/slurm/run_dppo_dev.sh
index 685c37f..33df3b7 100755
--- a/slurm/run_dppo_dev.sh
+++ b/slurm/run_dppo_dev.sh
@@ -17,9 +17,7 @@ module load devel/cuda/12.4
 # Set environment variables for WandB
 export WANDB_MODE=online
 export WANDB_PROJECT=dppo_dev_test
-export DPPO_WANDB_ENTITY=${DPPO_WANDB_ENTITY:-"your_wandb_username"}  # TODO: Set your WandB username
-# TODO: Set your WandB API key as environment variable:
-# export WANDB_API_KEY=<your_api_key>
+export DPPO_WANDB_ENTITY=${DPPO_WANDB_ENTITY:-"dominik_roth"}  # Use personal account, not shared
 
 # Default paths (can be overridden by environment)
 export DPPO_DATA_DIR=${DPPO_DATA_DIR:-$SLURM_SUBMIT_DIR/data}
diff --git a/slurm/run_dppo_gym.sh b/slurm/run_dppo_gym.sh
index 47b9e9f..dc447ab 100755
--- a/slurm/run_dppo_gym.sh
+++ b/slurm/run_dppo_gym.sh
@@ -14,11 +14,10 @@
 # Load required modules
 module load devel/cuda/12.4
 
-# Set environment variables
+# Set environment variables for WandB
 export WANDB_MODE=online
 export WANDB_PROJECT=dppo_gym
-# export WANDB_API_KEY=<your_api_key>  # TODO: Set your API key
-# export WANDB_ENTITY=<your_entity>    # TODO: Set your entity (username or team)
+export DPPO_WANDB_ENTITY=${DPPO_WANDB_ENTITY:-"dominik_roth"}  # Use personal account
 
 # Default paths (can be overridden by environment)
 export DPPO_DATA_DIR=${DPPO_DATA_DIR:-$SLURM_SUBMIT_DIR/data}