fix test for incorrect class attributes