ui bug fixed

Signed-off-by: Fei Guan <feiguan@172-16-186-178.dynapool.nyu.edu>