Fixed (I think) the updating of the state morphs.