Fixed broken integration test.