Make sure to wait at the end of the test