Fix demo bug