Fixed changing state visualization errors