fix ui bugs