Enhance util.wait()