lib/pelusa/lint/indentation_level.rb in pelusa-0.2.1 vs lib/pelusa/lint/indentation_level.rb in pelusa-0.2.2

@@ -27,14 +27,10 @@
           if node.is_a?(Rubinius::AST::Define)
             _iterate = Iterator.new do |node|
               __iterate = Iterator.new do |node|
                 if body = get_body_from_node[node]
                   if node.line != [body].flatten.first.line
-                    if body.is_a?(Array)
-                      @violations.merge body.map(&:line)
-                    else
-                      @violations << body.line
-                    end
+                    @violations.merge Array(body).map(&:line)
                   end
                 end
               end
 
               Array(get_body_from_node[node]).each(&__iterate)