add fix to tests according new UI