Fixed another bug in ui