Fix the class name for the tests