Small bugfix for one of the demo programs