fix ui #3