Fixed compilation error in demo