AI stun effect fix