Added `WaitForAttack` state