Fixed broken integration test case