lib/graphql/relay/mutation/instrumentation.rb in graphql-1.8.0.pre9 vs lib/graphql/relay/mutation/instrumentation.rb in graphql-1.8.0.pre10

@@ -8,10 +8,10 @@
         # in a {Mutation::Result}.
         #
         # By using an instrumention, we can apply our wrapper _last_,
         # giving users access to the original resolve function in earlier instrumentation.
         def self.instrument(type, field)
-          if field.mutation
+          if field.mutation.is_a?(GraphQL::Relay::Mutation) || (field.mutation.is_a?(Class) && field.mutation < GraphQL::Schema::RelayClassicMutation)
             new_resolve = Mutation::Resolve.new(field.mutation, field.resolve_proc)
             new_lazy_resolve = Mutation::Resolve.new(field.mutation, field.lazy_resolve_proc)
             field.redefine(resolve: new_resolve, lazy_resolve: new_lazy_resolve)
           else
             field