class Middleman::Extension

def bind_after_build
  ext = self
  if ext.respond_to?(:after_build)
    @klass.after_build do |builder|
      if ext.method(:after_build).arity == 1
        ext.after_build(builder)
      else
        ext.after_build
      end
    end
  end
end