Fixed compilation error for test class