Fixed failing integration test.