Small fix for the demo project