Fix callout UI tests