opal/virtual_dom.rb in opal-virtual-dom-0.0.1 vs opal/virtual_dom.rb in opal-virtual-dom-0.0.2

@@ -2,10 +2,10 @@
 require 'virtual_dom/virtual_text_node'
 require 'virtual_dom/node_factory'
 
 module VirtualDOM
   def virtual_dom(&block)
-    NodeFactory.new(block).nodes
+    NodeFactory.new(block, self).nodes.first
   end
 
   module_function
 
   def create(vnode)