diff --git a/opencompass/datasets/__init__.py b/opencompass/datasets/__init__.py
index 23cdf646d..568f3e2e0 100644
--- a/opencompass/datasets/__init__.py
+++ b/opencompass/datasets/__init__.py
@@ -35,7 +35,7 @@
 from .humanevalx import *  # noqa: F401, F403
 from .iwslt2017 import *  # noqa: F401, F403
 from .jigsawmultilingual import *  # noqa: F401, F403
-from .kaoshi import *  # noqa: F401, F403
+from .kaoshi import KaoshiDataset, KaoshiEvaluator  # noqa: F401, F403
 from .lambada import *  # noqa: F401, F403
 from .lcsts import *  # noqa: F401, F403
 from .LEval_coursera import *  # noqa: F401, F403