Fixed broken integration tests.