Fixed compilation error in test