Fix for integration test failing on nightly builds