Fix the UI tests/app