diff --git a/src/MaaCore/Task/Interface/RoguelikeTask.cpp b/src/MaaCore/Task/Interface/RoguelikeTask.cpp
index 1a54ac4733..81b0df9bde 100644
--- a/src/MaaCore/Task/Interface/RoguelikeTask.cpp
+++ b/src/MaaCore/Task/Interface/RoguelikeTask.cpp
@@ -131,18 +131,18 @@ bool asst::RoguelikeTask::set_params(const json::value& params)
         // 刷源石锭模式是否进入第二层
         bool investment_enter_second_floor = params.get("investment_enter_second_floor", true);
         if (investment_enter_second_floor) {
-            m_roguelike_task_ptr->set_times_limit("StageTraderLeave", INT_MAX);
+            m_roguelike_task_ptr->set_times_limit("StageTraderInvestCancel", INT_MAX);
             m_roguelike_task_ptr->set_times_limit("StageTraderLeaveConfirm", 0, ProcessTask::TimesLimitType::Post);
         }
         else {
-            m_roguelike_task_ptr->set_times_limit("StageTraderLeave", 0);
+            m_roguelike_task_ptr->set_times_limit("StageTraderInvestCancel", 0);
             m_roguelike_task_ptr->set_times_limit("StageTraderLeaveConfirm", INT_MAX);
         }
     }
     else {
         // 重置战斗后奖励next
         Task.set_task_base(theme + "@Roguelike@DropsFlag", theme + "@Roguelike@DropsFlag_default");
-        m_roguelike_task_ptr->set_times_limit("StageTraderLeave", INT_MAX);
+        m_roguelike_task_ptr->set_times_limit("StageTraderInvestCancel", INT_MAX);
         m_roguelike_task_ptr->set_times_limit("StageTraderLeaveConfirm", INT_MAX);
     }