Address Ameba issue

src/spectator/node_iterator.cr

@@ -18,25 +18,20 @@ module Spectator
     # Retrieves the next `Node`.
     # If there are none left, then `Iterator::Stop` is returned.
     def next
-      # Keep going until either:
+      # Nothing left to iterate.
-      # a. a node is found.
+      return stop if @stack.empty?
-      # b. the stack is empty.
-      until @stack.empty?
-        # Retrieve the next node.
-        node = @stack.pop
 
-        # If the node is a group, add its direct children to the queue
+      # Retrieve the next node.
-        # in reverse order so that the tree is traversed in pre-order.
+      node = @stack.pop
-        if node.is_a?(Indexable(Node))
-          node.reverse_each { |child| @stack.push(child) }
-        end
 
-        # Return the current node.
+      # If the node is a group, add its direct children to the queue
-        return node
+      # in reverse order so that the tree is traversed in pre-order.
+      if node.is_a?(Indexable(Node))
+        node.reverse_each { |child| @stack.push(child) }
       end
 
-      # Nothing left to iterate.
+      # Return the current node.
-      stop
+      node
     end
 
     # Restart the iterator at the beginning.