Make sure code with wait()s work