Fix UI test

see #243