fix broken integration test.