Update deprecated methods

The `set_state` and `remove_state` methods have been deprecated in
favour of `set_flag` and `clear_flag`. This updates the references to
those methods.