Fixing UI tests.