fix bug in  setup demo