ENH:fix UI