lib/lookbook/preview.rb in lookbook-2.0.0.beta.8 vs lib/lookbook/preview.rb in lookbook-2.0.0.beta.9

@@ -1,8 +1,9 @@
 module Lookbook
   class Preview
     include ActionView::Helpers::TagHelper
     include ActionView::Helpers::AssetTagHelper
+    extend PreviewAfterRender
 
     def render(component = nil, **args, &block)
       if component.nil?
         {
           type: :view,