diff --git a/src/you_get/extractors/iqiyi.py b/src/you_get/extractors/iqiyi.py
index 248446a7..0eea7d60 100644
--- a/src/you_get/extractors/iqiyi.py
+++ b/src/you_get/extractors/iqiyi.py
@@ -9,6 +9,7 @@ from math import floor
 from zlib import decompress
 import hashlib
 from ..util import log
+from .. import json_output
 
 import time