lib/mail/notify/mailers_controller.rb in mail-notify-1.0.5 vs lib/mail/notify/mailers_controller.rb in mail-notify-1.1.0

@@ -31,10 +31,10 @@
         @part = @email
         render action: "email", layout: false, formats: %i[html]
       end
 
       def notify?
-        @email.delivery_method.class == Mail::Notify::DeliveryMethod
+        @email.delivery_method.instance_of?(Mail::Notify::DeliveryMethod)
       end
     end
   end
 end