Fix UI error

see #127