diff --git a/pywb/static/wombat.js b/pywb/static/wombat.js
index 4af33bd8..b3008c85 100644
--- a/pywb/static/wombat.js
+++ b/pywb/static/wombat.js
@@ -2820,6 +2820,7 @@ var _WBWombat = function($wbwindow, wbinfo) {
             // override funcs to convert first arg proxy->obj
             override_func_first_arg_proxy_to_obj($wbwindow.MutationObserver.prototype, "observe");
             override_func_first_arg_proxy_to_obj($wbwindow.Node.prototype, "compareDocumentPosition");
+            override_func_first_arg_proxy_to_obj($wbwindow.Document.prototype, "createTreeWalker");
 
             override_frames_access($wbwindow);