fix broken UI test