Fix test by increasing maximum wait time