private void SetState(UnitState state) { if (currentState != null) { currentState.OnLeaveState(); } currentState = state; }