`wait_for_state` is more appropriate name for this methods. What do you think?