public void SetState(State state) { _currentState.EndState(); if (_battleStates.ContainsKey(state)) { _currentState = _battleStates[state]; } }