Modified the UI test