Fixed the broken integration test