Fixed an small bug in the demo.