Fixed deprecation warning for integration tests