Fixed include bug in demo