Revert mistake while asigning state classes