Fixed a problem with the demo