diff --git a/AI/BattleAI/AttackPossibility.h b/AI/BattleAI/AttackPossibility.h
index bab525c50..f370e87cc 100644
--- a/AI/BattleAI/AttackPossibility.h
+++ b/AI/BattleAI/AttackPossibility.h
@@ -9,7 +9,7 @@
  */
 #pragma once
 #include "../../lib/BattleState.h"
-#include "CCallback.h"
+#include "../../CCallback.h"
 #include "common.h"
 
 
diff --git a/AI/BattleAI/BattleAI.cbp b/AI/BattleAI/BattleAI.cbp
index 30f218952..60423bd42 100644
--- a/AI/BattleAI/BattleAI.cbp
+++ b/AI/BattleAI/BattleAI.cbp
@@ -66,12 +66,24 @@
 			<Add option="-lVCMI_lib" />
 			<Add directory="../.." />
 		</Linker>
+		<Unit filename="AttackPossibility.cpp" />
+		<Unit filename="AttackPossibility.h" />
 		<Unit filename="BattleAI.cpp" />
 		<Unit filename="BattleAI.h" />
+		<Unit filename="EnemyInfo.cpp" />
+		<Unit filename="EnemyInfo.h" />
+		<Unit filename="PotentialTargets.cpp" />
+		<Unit filename="PotentialTargets.h" />
+		<Unit filename="StackWithBonuses.cpp" />
+		<Unit filename="StackWithBonuses.h" />
 		<Unit filename="StdInc.h">
 			<Option compile="1" />
 			<Option weight="0" />
 		</Unit>
+		<Unit filename="ThreatMap.cpp" />
+		<Unit filename="ThreatMap.h" />
+		<Unit filename="common.cpp" />
+		<Unit filename="common.h" />
 		<Unit filename="main.cpp" />
 		<Extensions>
 			<code_completion />
diff --git a/AI/BattleAI/EnemyInfo.cpp b/AI/BattleAI/EnemyInfo.cpp
index 404e61f24..17ed65058 100644
--- a/AI/BattleAI/EnemyInfo.cpp
+++ b/AI/BattleAI/EnemyInfo.cpp
@@ -9,8 +9,8 @@
  */
 #include "StdInc.h"
 #include "EnemyInfo.h"
-#include "CRandomGenerator.h"
-#include "CCallback.h"
+#include "../../lib/CRandomGenerator.h"
+#include "../../CCallback.h"
 #include "common.h"
 
 void EnemyInfo::calcDmg(const CStack * ourStack)
diff --git a/AI/BattleAI/EnemyInfo.h b/AI/BattleAI/EnemyInfo.h
index 085dd793c..52212f050 100644
--- a/AI/BattleAI/EnemyInfo.h
+++ b/AI/BattleAI/EnemyInfo.h
@@ -8,7 +8,7 @@
  *
  */
 #pragma once
-#include "BattleHex.h"
+#include "../../lib/BattleState.h"
 
 class CStack;