diff --git a/FlagEmbedding/finetune/embedder/decoder_only/icl/__init__.py b/FlagEmbedding/finetune/embedder/decoder_only/icl/__init__.py
index 4863ce4..57ec78e 100644
--- a/FlagEmbedding/finetune/embedder/decoder_only/icl/__init__.py
+++ b/FlagEmbedding/finetune/embedder/decoder_only/icl/__init__.py
@@ -6,6 +6,10 @@ from .arguments import (
     DecoderOnlyEmbedderICLModelArguments,
     DecoderOnlyEmbedderICLDataArguments
 )
+from .dataset import (
+    DecoderOnlyEmbedderICLSameDatasetTrainDataset,
+    AbsEmbedderSameDatasetCollator
+)
 from .modeling import BiDecoderOnlyEmbedderICLModel
 from .trainer import DecoderOnlyEmbedderICLTrainer
 from .runner import DecoderOnlyEmbedderICLRunner