1 files changed, 3 insertions, 0 deletions
diff --git a/panfry/main.py b/panfry/main.py
index 40c2b15..9572243 100755
--- a/panfry/main.py
+++ b/panfry/main.py
@@ -62,6 +62,9 @@ def gen(env):
     if env.simple_toc:
         document.set_simple_toc
 
+    if env.json_toc:
+        document.set_json_toc
+
     if env.clean and os.path.exists(env.pub_path):
         print("Cleanup, removing %s..." % env.pub_path)
         shutil.rmtree(env.pub_path)