Fixing deprecation warnings on the integration tests