lib/pessimize/gemfile.rb in pessimize-0.3.0 vs lib/pessimize/gemfile.rb in pessimize-0.4.0

@@ -40,10 +40,10 @@
       end
     rescue StopIteration
     end
 
     def all_gem_tokens_collected?(current_token, next_token)
-      next_token[1] == :on_nl
+      [:on_nl, :on_comment].include?(next_token[1])
     end
 
     class TokenCompiler
       def initialize(tokens)
         self.tokens = tokens.dup