Fix for UI tests.