a little fix for the demo