Fixing UI tests, updating UI test screenshot.