Fixing issue with console errors in the demo