diff --git a/lib/banzai/filter/references/reference_filter.rb b/lib/banzai/filter/references/reference_filter.rb
index 41d2220498df59ca646fb87b5b0a511b48dfe5f0..48f718c9d6ee086fedd3c3bb730015834d582705 100644
--- a/lib/banzai/filter/references/reference_filter.rb
+++ b/lib/banzai/filter/references/reference_filter.rb
@@ -348,24 +348,8 @@ def query
         def replace_node_with_html(node, index, html)
           return if node.to_html == html
 
-          previous_node = node.previous
-          next_node = node.next
-          parent_node = node.parent
-          # Unfortunately node.replace(html) returns re-parented nodes, not the actual replaced nodes in the doc
-          # We need to find the actual nodes in the doc that were replaced
-          node.replace(html)
-          @new_nodes[index] = []
-
-          # We replaced node with new nodes, so we find first new node. If previous_node is nil, we take first parent child
-          new_node = previous_node ? previous_node.next : parent_node&.children&.first
-
-          # We iterate from first to last replaced node and store replaced nodes in @new_nodes
-          while new_node && new_node != next_node
-            @new_nodes[index] << new_node.xpath(query)
-            new_node = new_node.next
-          end
-
-          @new_nodes[index].flatten!
+          created = node.replace(html)
+          @new_nodes[index] = created.xpath(query).to_a
         end
 
         def only_path?