lib/rouge/lexers/gherkin.rb in rouge-1.1.0 vs lib/rouge/lexers/gherkin.rb in rouge-1.2.0

@@ -52,20 +52,20 @@
       state :has_scenarios do
         rule %r((.*?)(:)) do |m|
           reset_stack
 
           keyword = m[1]
-          if self.class.keywords[:element].include? keyword
-            group Keyword::Namespace; push :description
+          keyword_tok = if self.class.keywords[:element].include? keyword
+            push :description; Keyword::Namespace
           elsif self.class.keywords[:feature].include? keyword
-            group Keyword::Declaration; push :feature_description
+            push :feature_description; Keyword::Declaration
           elsif self.class.keywords[:examples].include? keyword
-            group Name::Namespace; push :example_description
+            push :example_description; Name::Namespace
           else
-            group Error
+            Error
           end
 
-          group Punctuation
+          groups keyword_tok, Punctuation
         end
       end
 
       state :has_examples do
         mixin :has_scenarios