fix error in test class