Bug fix

[X]  Some units would get stuck in the attack phase, this update fixes
that bug.