Fix broken test case in new module.