UI: Fix UI tests