UI Test correction