fixed demo after api enhancement