fixed test class errors