Test fix for recent UI changes