Fix tests to use the right module name