fix wait_on