[M] Fix demo bug