Update test class names