Unit Test Fixes UI