Increase wait time in test