Fix class name in tests.