lib/middleman-cdn/extension.rb in middleman-cdn-0.1.1 vs lib/middleman-cdn/extension.rb in middleman-cdn-0.1.2

@@ -15,16 +15,18 @@
       option :after_build, false, 'Cloudflare options'
 
       def initialize(app, options_hash = {}, &block)
         super
 
-        app.after_build do
-          ::Middleman::Cli::CDN.new.invalidate(options) if options.after_build
+        @@cdn_options = options
+
+        app.after_configuration do
+          app.after_build do
+            ::Middleman::Cli::CDN.new.invalidate(@@cdn_options) if @@cdn_options.after_build
+          end
         end
 
         app.send :include, Helpers
-
-        @@cdn_options = options
       end
 
       def registered
         included
       end