Update test to cope with changed UI class (#10867)

Change-Id: I6e822c6e99191a68733c87207f61643213e8c154