Fixed compilation error in tests