Small fix for demo app