diff --git a/opencompass/tasks/openicl_attack.py b/opencompass/tasks/openicl_attack.py
index 09190dc7..773fce74 100644
--- a/opencompass/tasks/openicl_attack.py
+++ b/opencompass/tasks/openicl_attack.py
@@ -15,7 +15,7 @@ from opencompass.utils import (build_dataset_from_cfg, build_model_from_cfg,
                                task_abbr_from_cfg)
 
 
-@TASKS.register_module(force=(__name__ == '__main__'))  # A hack for script run
+@TASKS.register_module()
 class OpenICLAttackTask(BaseTask):
     """OpenICL Inference Task.
 
diff --git a/opencompass/tasks/openicl_eval.py b/opencompass/tasks/openicl_eval.py
index 8d2494d8..1fd9e7ed 100644
--- a/opencompass/tasks/openicl_eval.py
+++ b/opencompass/tasks/openicl_eval.py
@@ -55,7 +55,7 @@ def extract_role_pred(s: str, begin_str: Optional[str],
     return s[start:end]
 
 
-@TASKS.register_module(force=(__name__ == '__main__'))  # A hack for script run
+@TASKS.register_module()
 class OpenICLEvalTask(BaseTask):
     """OpenICL Evaluation Task.
 
diff --git a/opencompass/tasks/openicl_infer.py b/opencompass/tasks/openicl_infer.py
index a4a83190..0157cbcf 100644
--- a/opencompass/tasks/openicl_infer.py
+++ b/opencompass/tasks/openicl_infer.py
@@ -17,7 +17,7 @@ from opencompass.utils import (build_dataset_from_cfg, build_model_from_cfg,
                                model_abbr_from_cfg, task_abbr_from_cfg)
 
 
-@TASKS.register_module(force=(__name__ == '__main__'))  # A hack for script run
+@TASKS.register_module()
 class OpenICLInferTask(BaseTask):
     """OpenICL Inference Task.