tried to fix test class