diff --git a/source/Image.py b/source/Image.py
index fd682d4..a7804d4 100644
--- a/source/Image.py
+++ b/source/Image.py
@@ -8,7 +8,7 @@ class Image:
     def __init__(self, html):
         self.html = html
 
-    def get_image_link(self, url: str, download: bool):
+    def get_image_link(self, url: str):
         html = self.html.get_html(url)
         return self.__get_image_links(html)
 
diff --git a/source/Video.py b/source/Video.py
index 61574b3..2c9accd 100644
--- a/source/Video.py
+++ b/source/Video.py
@@ -7,7 +7,7 @@ class Video:
     def __init__(self, html):
         self.html = html
 
-    def get_video_link(self, url: str, download: bool):
+    def get_video_link(self, url: str):
         html = self.html.get_html(url)
         return self.__get_video_link(html)
 
diff --git a/source/__init__.py b/source/__init__.py
index f8d7913..05e0603 100644
--- a/source/__init__.py
+++ b/source/__init__.py
@@ -18,7 +18,13 @@ class XHS:
         self._cookie = ""
 
     def get_image(self, url: str, download=False):
-        return self.image.get_image_link(url, download)
+        urls = self.image.get_image_link(url)
+        if download:
+            pass
+        return urls
 
     def get_video(self, url: str, download=False):
-        return self.video.get_video_link(url, download)
+        url = self.video.get_video_link(url)
+        if download:
+            pass
+        return url