diff --git a/pywb/rewrite/rewrite_live.py b/pywb/rewrite/rewrite_live.py
index 60ab848b..b3e2d464 100644
--- a/pywb/rewrite/rewrite_live.py
+++ b/pywb/rewrite/rewrite_live.py
@@ -284,7 +284,6 @@ class YoutubeDLWrapper(object):  #pragma: no cover
         self.ydl.add_default_info_extractors()
 
     def extract_info(self, url):
-        print('YDL', self.ydl)
         if not self.ydl:
             return None
 
diff --git a/tests/test_live_proxy.py b/tests/test_live_proxy.py
index 7896eac5..5ef008b1 100644
--- a/tests/test_live_proxy.py
+++ b/tests/test_live_proxy.py
@@ -10,7 +10,7 @@ from pywb.framework.wsgi_wrappers import init_app
 import webtest
 import shutil
 
-import pywb.webapp.live_rewrite_handler
+import pywb.rewrite.rewrite_live
 
 
 #=================================================================
@@ -53,7 +53,7 @@ class MockYTDWrapper(object):
         return {'mock': 'youtube_dl_data'}
 
 
-pywb.webapp.live_rewrite_handler.YoutubeDLWrapper = MockYTDWrapper
+pywb.rewrite.rewrite_live.youtubedl = MockYTDWrapper()
 
 
 #=================================================================
diff --git a/tests/test_live_rewriter.py b/tests/test_live_rewriter.py
index 623dfe34..be904d68 100644
--- a/tests/test_live_rewriter.py
+++ b/tests/test_live_rewriter.py
@@ -2,7 +2,7 @@ from pywb.webapp.live_rewrite_handler import RewriteHandler
 from pywb.apps.cli import LiveCli
 from pywb.framework.wsgi_wrappers import init_app
 import webtest
-import pywb.webapp.live_rewrite_handler
+import pywb.rewrite.rewrite_live
 
 
 #=================================================================
@@ -11,7 +11,7 @@ class MockYTDWrapper(object):
         return {'mock': 'youtube_dl_data'}
 
 
-pywb.webapp.live_rewrite_handler.YoutubeDLWrapper = MockYTDWrapper
+pywb.rewrite.rewrite_live.youtubedl = MockYTDWrapper()
 
 
 def setup_module():