modify test class