diff --git a/scripts/utils.py b/scripts/utils.py
index 2a84a8d..49c5a89 100644
--- a/scripts/utils.py
+++ b/scripts/utils.py
@@ -3,7 +3,7 @@
 
 import fastchat
 import transformers
-from dataset import Dataset, load_dataset
+from datasets import Dataset, load_dataset
 
 from rewardbench.utils import check_tokenizer_chat_template, prepare_dialogue
 from rewardbench.utils import prepare_dialogue_from_tokenizer