Fix integration test