diff --git a/opencompass/configs/datasets/aime2024/aime2024_llmjudge_gen_5e9f4f.py b/opencompass/configs/datasets/aime2024/aime2024_llmjudge_gen_5e9f4f.py
index c1be29dd..a54e53f1 100644
--- a/opencompass/configs/datasets/aime2024/aime2024_llmjudge_gen_5e9f4f.py
+++ b/opencompass/configs/datasets/aime2024/aime2024_llmjudge_gen_5e9f4f.py
@@ -1,7 +1,7 @@
 from opencompass.openicl.icl_prompt_template import PromptTemplate
 from opencompass.openicl.icl_retriever import ZeroRetriever
 from opencompass.openicl.icl_inferencer import GenInferencer
-from opencompass.datasets import CustomDataset
+from opencompass.datasets import Aime2024Dataset
 from opencompass.evaluator import GenericLLMEvaluator
 from opencompass.datasets import generic_llmjudge_postprocess
 
@@ -69,7 +69,7 @@ aime2024_eval_cfg = dict(
             ),
         ),
         dataset_cfg=dict(
-            type=CustomDataset,
+            type=Aime2024Dataset,
             path='opencompass/aime2024',
             reader_cfg=aime2024_reader_cfg,
         ),
@@ -81,7 +81,7 @@ aime2024_eval_cfg = dict(
 aime2024_datasets = [
     dict(
         abbr='aime2024',
-        type=CustomDataset,
+        type=Aime2024Dataset,
         path='opencompass/aime2024',
         reader_cfg=aime2024_reader_cfg,
         infer_cfg=aime2024_infer_cfg,