Update to handle the modules being in the root directory.