Fixed the test class names