lib/jwe/alg.rb in jwe-0.1.0 vs lib/jwe/alg.rb in jwe-0.1.1

@@ -4,10 +4,10 @@
 
 module JWE
   module Alg
     def self.for(alg)
       klass = alg.gsub(/[-\+]/, '_').downcase.sub(/^[a-z\d]*/) { $&.capitalize }
-      klass.gsub!(/_([a-z\d]*)/i) { $1.capitalize }
+      klass.gsub!(/_([a-z\d]*)/i) { Regexp.last_match(1).capitalize }
       const_get(klass)
 
     rescue NameError
       raise NotImplementedError.new("Unsupported alg type: #{alg}")
     end