change the test class for test