fix demo compilation issue