Fix broken integration test