small bug fixed in ui