Make sure code with wait()s work

Make sure code with wait()s work