Fixing demo to work :)