setting the remaining changes caused by the change to module.wait