Updated ui tests