Updates to integration tests.