Support wait_time parameter