Fixed  bug in custom UI demo.