diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index ed9d26b..bd159f2 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -27,8 +27,9 @@ jobs:
         pytest
     - name: Check if cog needs to be run
       run: |
-        cog docs/usage.md
-        cog --check docs/*.md
+        cog --check \
+          -p "import sys, os; sys._called_from_test=True; os.environ['LLM_USER_PATH'] = '/tmp'" \
+          docs/*.md
     - name: Run Black
       run: |
         black --check .
diff --git a/Justfile b/Justfile
index d83f403..ba2f05f 100644
--- a/Justfile
+++ b/Justfile
@@ -23,7 +23,7 @@
 
 # Rebuild docs with cog
 @cog:
-  pipenv run cog -r docs/*.md
+  pipenv run cog -r -p "import sys, os; sys._called_from_test=True; os.environ['LLM_USER_PATH'] = '/tmp'" docs/*.md
 
 # Serve live docs on localhost:8000
 @docs: cog
diff --git a/docs/aliases.md b/docs/aliases.md
index 869dd78..d3ef1b4 100644
--- a/docs/aliases.md
+++ b/docs/aliases.md
@@ -12,9 +12,6 @@ Example output:
 
 <!-- [[[cog
 from click.testing import CliRunner
-import sys, os
-os.environ["LLM_USER_PATH"] = "/tmp"
-sys._called_from_test = True
 from llm.cli import cli
 result = CliRunner().invoke(cli, ["aliases", "list"])
 cog.out("```\n{}```".format(result.output))
diff --git a/docs/help.md b/docs/help.md
index 3d7f634..f58cd66 100644
--- a/docs/help.md
+++ b/docs/help.md
@@ -4,8 +4,6 @@ This page lists the `--help` output for all of the `llm` commands.
 
 <!-- [[[cog
 from click.testing import CliRunner
-import sys
-sys._called_from_test = True
 from llm.cli import cli
 def all_help(cli):
     "Return all help for Click command and its subcommands"
diff --git a/docs/usage.md b/docs/usage.md
index 67c71ec..278550a 100644
--- a/docs/usage.md
+++ b/docs/usage.md
@@ -90,12 +90,8 @@ llm models list --options
 Output:
 <!-- [[[cog
 from click.testing import CliRunner
-import os, sys
-sys._called_from_test = True
 from llm.cli import cli
-os.environ["LLM_USER_PATH"] = "/tmp"
 result = CliRunner().invoke(cli, ["models", "list", "--options"])
-del os.environ["LLM_USER_PATH"]
 cog.out("```\n{}\n```".format(result.output))
 ]]] -->
 ```