Updates with the correct module name for import