fix unit test error
ui fix