Fixing ui tests.