Why do you no longer do this self._ok_states check?