lib/octopress-docs/hooks.rb in octopress-docs-0.0.10 vs lib/octopress-docs/hooks.rb in octopress-docs-0.0.11

@@ -1,21 +1,21 @@
 module Octopress
   module Docs
     class DocsSiteHook < Octopress::Hooks::Site
       def post_read(site)
-        if Octopress::Docs.docs_mode && Octopress.site.nil?
+        if ENV['OCTOPRESS_DOCS'] && Octopress.site.nil?
           Octopress.site = site
         end
       end
       
       def pre_render(site)
-        if Octopress::Docs.docs_mode
+        if ENV['OCTOPRESS_DOCS']
           site.pages.concat Octopress::Docs.pages
         end
       end
 
       def merge_payload(payload, site)
-        if Octopress::Docs.docs_mode
+        if ENV['OCTOPRESS_DOCS']
           Octopress::Docs.pages_info
         end
       end
     end
   end