Add .wait_for_present to set_field for checkbox