Can we immediately assign `newState` to the extend here?