Fixed small quirk in the demo