fix test class!