Update for ui test