Minor customization to test