Dialogue diagram breaks when a state is removed