Fix bugs with late state update. Add Message.copy.