Fix to integration test