Merge pull request #170 from Pistonight/fix/new_viking_mismatches

take new viking and update mismatches

1 files changed, 2 insertions, 2 deletions

diff --git a/src/Game/AI/Query/queryIsCurrentMainFieldStage.cpp b/src/Game/AI/Query/queryIsCurrentMainFieldStage.cpp
index ddd0aa88..879bfced 100644
--- a/src/Game/AI/Query/queryIsCurrentMainFieldStage.cpp
+++ b/src/Game/AI/Query/queryIsCurrentMainFieldStage.cpp
@@ -1,6 +1,6 @@
 #include "Game/AI/Query/queryIsCurrentMainFieldStage.h"
 #include <evfl/Query.h>
-#include "KingSystem/System/StageInfo.h"
+#include "Game/gameScene.h"
 
 namespace uking::query {
 
@@ -9,7 +9,7 @@ IsCurrentMainFieldStage::IsCurrentMainFieldStage(const InitArg& arg) : ksys::act
 IsCurrentMainFieldStage::~IsCurrentMainFieldStage() = default;
 
 int IsCurrentMainFieldStage::doQuery() {
-    return ksys::StageInfo::getCurrentMapType() == "MainField";
+    return GameScene::getCurrentMapType() == "MainField";
 }
 
 void IsCurrentMainFieldStage::loadParams(const evfl::QueryArg& arg) {}