Fixed broken integration test