Fix the test build (UI tests still broken)