Fixing uI tests.