diff --git a/evaluation.py b/evaluation.py
index 2e3f196..a667ce9 100644
--- a/evaluation.py
+++ b/evaluation.py
@@ -1,5 +1,6 @@
 import random
 import torch
+import wandb
 
 
 def generate_passkey_prompt(passkey, context_length):