registry/unit_state: fix tests