Maybe this fixes the UI tests