diff --git a/README.md b/README.md
index 2b685d8..279a50b 100644
--- a/README.md
+++ b/README.md
@@ -18,7 +18,11 @@ Put something like this in your shell's config, e.g. `.bashrc`, `.zshrc`, or wha
 
 ```bash
 function dig() {
-    /usr/bin/dig $@ | pygmentize -xl path/to/dig-lexer.py -Ostyle=monokai
+    if [[ -t 1 || "$DIG_COLOR" == "always" ]]; then
+        /usr/bin/dig $@ | pygmentize -F tokenmerge -xl path/to/dig-lexer.py -Ostyle=monokai
+    else
+        /usr/bin/dig $@
+    fi
 }
 ```