1 files changed, 15 insertions, 1 deletions
diff --git a/src/game/IdleMovementGenerator.cpp b/src/game/IdleMovementGenerator.cpp
index 2428c49a77a..44575302e75 100644
--- a/src/game/IdleMovementGenerator.cpp
+++ b/src/game/IdleMovementGenerator.cpp
@@ -19,7 +19,8 @@
  */
 
 #include "IdleMovementGenerator.h"
-#include "Unit.h"
+#include "CreatureAI.h"
+#include "Creature.h"
 
 IdleMovementGenerator si_idleMovement;
 
@@ -60,3 +61,16 @@ DistractMovementGenerator::Update(Unit& owner, const uint32& time_diff)
     return true;
 }
 
+void
+AssistanceDistractMovementGenerator::Finalize(Unit &unit)
+{
+    unit.clearUnitState(UNIT_STAT_DISTRACTED);
+    if (Unit* victim = unit.getVictim())
+    {
+        if (unit.isAlive())
+        {
+            unit.AttackStop();
+            ((Creature*)&unit)->AI()->AttackStart(victim);
+        }
+    }
+}