fixed test class