Fix broken UITests