Fixing demo
Minor change