fixes in the demo