I think to do this, you must have a '==' defined on the class (or decorator if you go that approach). Otherwise, these states will not be '==', or are these strings here?