Fix ui tests