diff --git a/run.py b/run.py
index 728bea4..5212bc0 100644
--- a/run.py
+++ b/run.py
@@ -52,7 +52,7 @@ logger = logging.getLogger("desktopenv.experiment")
 
 # wandb config
 ### set your wandb api key here
-wandb.login(key=os.get("WANDB_API_KEY", None))
+wandb.login(key=os.environ.get("WANDB_API_KEY", None))
 
 
 def config() -> argparse.Namespace: