test class fix