small bug fix (ui)