diff --git a/evaluation/chat_humaneval.py b/evaluation/chat_humaneval.py
index 0abe109..c9ed111 100755
--- a/evaluation/chat_humaneval.py
+++ b/evaluation/chat_humaneval.py
@@ -25,7 +25,7 @@
 
 from evaluate import load
 
-from lm_eval.base import Task
+from bigcode_eval.base import Task
 
 _CITATION = """
 @misc{chen2021evaluating,