Update ui tests