Fix UI tests.