diff --git a/opencompass/datasets/rbench.py b/opencompass/datasets/rbench.py
index ddbddc6b..a2f03950 100644
--- a/opencompass/datasets/rbench.py
+++ b/opencompass/datasets/rbench.py
@@ -1,4 +1,4 @@
-from datasets import Dataset, DatasetDict, load_dataset
+from datasets import Dataset, load_dataset
 from opencompass.registry import LOAD_DATASET
 from .base import BaseDataset