Minor change on the demo