Fix ui tests.