Small fix for demo