diff --git a/reme_ai/react/__init__.py b/reme_ai/react/__init__.py
index e69de29b..f49ce6d5 100644
--- a/reme_ai/react/__init__.py
+++ b/reme_ai/react/__init__.py
@@ -0,0 +1 @@
+from .react_v1_op import ReactV1Op
diff --git a/reme_ai/retrieve/__init__.py b/reme_ai/retrieve/__init__.py
index 53a36c1f..0d054f82 100644
--- a/reme_ai/retrieve/__init__.py
+++ b/reme_ai/retrieve/__init__.py
@@ -1,2 +1,2 @@
-from reme_ai.retrieve import task
-from reme_ai.retrieve import personal
+from . import task
+from . import personal
diff --git a/reme_ai/retrieve/task/__init__.py b/reme_ai/retrieve/task/__init__.py
index 88fed8b8..cf6c1ded 100644
--- a/reme_ai/retrieve/task/__init__.py
+++ b/reme_ai/retrieve/task/__init__.py
@@ -1,2 +1,4 @@
-from reme_ai.retrieve.task.build_query_op import BuildQueryOp
-from reme_ai.retrieve.task.merge_memory_op import MergeMemoryOp
\ No newline at end of file
+from .build_query_op import BuildQueryOp
+from .merge_memory_op import MergeMemoryOp
+from .rerank_memory_op import RerankMemoryOp
+from .rewrite_memory_op import RewriteMemoryOp
diff --git a/reme_ai/summary/__init__.py b/reme_ai/summary/__init__.py
index 1f47c200..dec8dbf7 100644
--- a/reme_ai/summary/__init__.py
+++ b/reme_ai/summary/__init__.py
@@ -1,2 +1,2 @@
-from reme_ai.summary import task
-from reme_ai.summary import personal
\ No newline at end of file
+from . import task
+from . import personal
\ No newline at end of file
diff --git a/reme_ai/summary/task/__init__.py b/reme_ai/summary/task/__init__.py
index c6b143af..169f8958 100644
--- a/reme_ai/summary/task/__init__.py
+++ b/reme_ai/summary/task/__init__.py
@@ -1 +1,10 @@
-from reme_ai.summary.task.simple_summary_op import SimpleSummaryOp
\ No newline at end of file
+from .comparative_extraction_op import ComparativeExtractionOp
+from .failure_extraction_op import FailureExtractionOp
+from .memory_deduplication_op import MemoryDeduplicationOp
+from .memory_validation_op import MemoryValidationOp
+from .pdf_preprocess_op_wrapper import PDFPreprocessOp
+from .simple_comparative_summary_op import SimpleComparativeSummaryOp
+from .simple_summary_op import SimpleSummaryOp
+from .success_extraction_op import SuccessExtractionOp
+from .trajectory_preprocess_op import TrajectoryPreprocessOp
+from .trajectory_segmentation_op import TrajectorySegmentationOp
diff --git a/reme_ai/summary/task/memory_deduplication_op.py b/reme_ai/summary/task/memory_deduplication_op.py
index 8982a576..a8d33266 100644
--- a/reme_ai/summary/task/memory_deduplication_op.py
+++ b/reme_ai/summary/task/memory_deduplication_op.py
@@ -6,7 +6,7 @@ from reme_ai.schema.memory import BaseMemory
 
 
 @C.register_op()
-class TaskMemoryDeduplicationOp(BaseOp):
+class MemoryDeduplicationOp(BaseOp):
     file_path: str = __file__
 
     def execute(self):