Fixed demo bug