adding wait to the test