This wait seems almost the same as the one below it. Can we remove this wait? The test works without it