lib/mongo/operation/shared/write.rb in mongo-2.15.1 vs lib/mongo/operation/shared/write.rb in mongo-2.16.0.alpha1

@@ -33,10 +33,10 @@
       #
       # @return [ Mongo::Operation::Result ] The operation result.
       #
       # @since 2.5.2
       def execute(server, context:)
-        server.with_connection do |connection|
+        server.with_connection(service_id: context.service_id) do |connection|
           validate!(connection)
           op = if connection.features.op_msg_enabled?
               self.class::OpMsg.new(spec)
             elsif !acknowledged_write?
               self.class::Legacy.new(spec)