modify test for class