diff --git a/rdagent/app/data_science/conf.py b/rdagent/app/data_science/conf.py
index 8b5330ab..ebdd4db6 100644
--- a/rdagent/app/data_science/conf.py
+++ b/rdagent/app/data_science/conf.py
@@ -84,7 +84,7 @@ class DataScienceBasePropSetting(KaggleBasePropSetting):
     """The maximum number of traces to grow before merging"""
 
     #### multi-trace:checkpoint selector
-    selector_name: str = "rdagent.scenarios.data_science.proposal.exp_gen.ckp_select.LatestCKPSelector"
+    selector_name: str = "rdagent.scenarios.data_science.proposal.exp_gen.select.expand.LatestCKPSelector"
     """The name of the selector to use"""
     sota_count_window: int = 5
     """The number of trials to consider for SOTA count"""
@@ -92,7 +92,7 @@ class DataScienceBasePropSetting(KaggleBasePropSetting):
     """The threshold for SOTA count"""
 
     #### multi-trace: SOTA experiment selector
-    sota_exp_selector_name: str = "rdagent.scenarios.data_science.proposal.exp_gen.sota_exp_select.GlobalSOTASelector"
+    sota_exp_selector_name: str = "rdagent.scenarios.data_science.proposal.exp_gen.select.submit.GlobalSOTASelector"
     """The name of the SOTA experiment selector to use"""
 
     ### multi-trace:inject optimals for multi-trace
diff --git a/rdagent/oai/backend/litellm.py b/rdagent/oai/backend/litellm.py
index b5d1e759..1f9574bb 100644
--- a/rdagent/oai/backend/litellm.py
+++ b/rdagent/oai/backend/litellm.py
@@ -98,7 +98,7 @@ class LiteLLMAPIBackend(APIBackend):
         if response_format and not supports_response_schema(model=LITELLM_SETTINGS.chat_model):
             # Deepseek will enter this branch
             logger.warning(
-                f"{LogColors.RED}Model {LITELLM_SETTINGS.chat_model} does not support response schema, ignoring response_format argument.{LogColors.END}",
+                f"{LogColors.YELLOW}Model {LITELLM_SETTINGS.chat_model} does not support response schema, ignoring response_format argument.{LogColors.END}",
                 tag="llm_messages",
             )
             response_format = None
diff --git a/rdagent/scenarios/data_science/proposal/exp_gen/ckp_select.py b/rdagent/scenarios/data_science/proposal/exp_gen/select/expand.py
similarity index 100%
rename from rdagent/scenarios/data_science/proposal/exp_gen/ckp_select.py
rename to rdagent/scenarios/data_science/proposal/exp_gen/select/expand.py
diff --git a/rdagent/scenarios/data_science/proposal/exp_gen/prompts_selector.yaml b/rdagent/scenarios/data_science/proposal/exp_gen/select/prompts.yaml
similarity index 100%
rename from rdagent/scenarios/data_science/proposal/exp_gen/prompts_selector.yaml
rename to rdagent/scenarios/data_science/proposal/exp_gen/select/prompts.yaml
diff --git a/rdagent/scenarios/data_science/proposal/exp_gen/sota_exp_select.py b/rdagent/scenarios/data_science/proposal/exp_gen/select/submit.py
similarity index 96%
rename from rdagent/scenarios/data_science/proposal/exp_gen/sota_exp_select.py
rename to rdagent/scenarios/data_science/proposal/exp_gen/select/submit.py
index e8875f17..f478d7c2 100644
--- a/rdagent/scenarios/data_science/proposal/exp_gen/sota_exp_select.py
+++ b/rdagent/scenarios/data_science/proposal/exp_gen/select/submit.py
@@ -115,11 +115,9 @@ class AutoSOTAexpSelector(SOTAexpSelector):
                         Description: {desc}
                         Final score: {current_final_score}\n\n"""
 
-            system_prompt = T(".prompts_selector:auto_sota_selector.system").r(
-                scenario=trace.scen.get_scenario_all_desc()
-            )
+            system_prompt = T(".prompts:auto_sota_selector.system").r(scenario=trace.scen.get_scenario_all_desc())
 
-            user_prompt = T(".prompts_selector:auto_sota_selector.user").r(
+            user_prompt = T(".prompts:auto_sota_selector.user").r(
                 historical_sota_exp_with_desc_and_scores=SOAT_exp_with_desc_and_scores,
             )