Update StunState.java