X-Git-Url: https://jfr.im/git/yt-dlp.git/blobdiff_plain/8c5fee2ea5b98b233062058de14b90dfa5bb9f0e..cc52de43568d8cd58c7e2ef4e5cecf609da28a9c:/test/test_write_annotations.py

diff --git a/test/test_write_annotations.py b/test/test_write_annotations.py
index d98c96c15..fa31be0cc 100644
--- a/test/test_write_annotations.py
+++ b/test/test_write_annotations.py
@@ -1,4 +1,4 @@
-#!/usr/bin/env python
+#!/usr/bin/env python3
 # coding: utf-8
 from __future__ import unicode_literals
 
@@ -15,11 +15,11 @@
 
 import xml.etree.ElementTree
 
-import youtube_dlc.YoutubeDL
-import youtube_dlc.extractor
+import yt_dlp.YoutubeDL
+import yt_dlp.extractor
 
 
-class YoutubeDL(youtube_dlc.YoutubeDL):
+class YoutubeDL(yt_dlp.YoutubeDL):
     def __init__(self, *args, **kwargs):
         super(YoutubeDL, self).__init__(*args, **kwargs)
         self.to_stderr = self.to_screen
@@ -45,7 +45,7 @@ def setUp(self):
 
     def test_info_json(self):
         expected = list(EXPECTED_ANNOTATIONS)  # Two annotations could have the same text.
-        ie = youtube_dlc.extractor.YoutubeIE()
+        ie = yt_dlp.extractor.YoutubeIE()
         ydl = YoutubeDL(params)
         ydl.add_info_extractor(ie)
         ydl.download([TEST_ID])