Updated to work with new module name.