Ui test fixes