Fix failing IT tests