wait_until has been deprecated so using Timeout
instead.