diff --git a/model/reward/instructor/cls_dataset.py b/model/reward/instructor/cls_dataset.py
index 09aa821b..7992c37c 100644
--- a/model/reward/instructor/cls_dataset.py
+++ b/model/reward/instructor/cls_dataset.py
@@ -8,7 +8,7 @@ import json
 import os
 import random
 
-from dataset import load_dataset
+from datasets import load_dataset
 from torch.utils.data import Dataset
 
 from .utils import webgpt_return_format