diff --git a/test/test_post_hooks.py b/test/test_post_hooks.py index d8d2b36c3..63500924e 100644 --- a/test/test_post_hooks.py +++ b/test/test_post_hooks.py @@ -8,11 +8,11 @@ import unittest sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) from test.helper import get_params, try_rm -import youtube_dl.YoutubeDL -from youtube_dl.utils import DownloadError +import youtube_dlc.YoutubeDL +from youtube_dlc.utils import DownloadError -class YoutubeDL(youtube_dl.YoutubeDL): +class YoutubeDL(youtube_dlc.YoutubeDL): def __init__(self, *args, **kwargs): super(YoutubeDL, self).__init__(*args, **kwargs) self.to_stderr = self.to_screen