fix unit tests in UI module