diff --git a/src/game/UnitAI.cpp b/src/game/UnitAI.cpp
index 76460bd8cc..12dc20692c 100644
--- a/src/game/UnitAI.cpp
+++ b/src/game/UnitAI.cpp
@@ -19,7 +19,8 @@
  */
 
 #include "UnitAI.h"
-#include "Unit.h"
+#include "Player.h"
+#include "Creature.h"
 #include "SpellAuras.h"
 
 void UnitAI::AttackStart(Unit *victim)