Ensure that state diagrams are correctly refreshed after a step operation