Fix stun crash