Update tests to reflect module name.