Fixed compilation problem