Fixed an issue with an integration test.