Fixed failing tests by correcting the module name