fix some bug about UI