add changes for ui tests