Fix: a UI bug