Fixed failing UI Test