Fixed dumb bug in the demo :)