fix test class