add test for _update_rule_for_states