Fix broken integration tests.