diff --git a/src/api/src/create_book/__init__.py b/src/api/src/create_book/__init__.py
index ef54aa5..9f74c06 100644
--- a/src/api/src/create_book/__init__.py
+++ b/src/api/src/create_book/__init__.py
@@ -4,10 +4,10 @@ from .create_book import (
     fetch_story,
     fetch_story_from_partId,
     fetch_story_content_zip,
-    fetch_image,
     fetch_cookies,
 )
 from .generators import PDFGenerator, EPUBGenerator
 from .exceptions import WattpadError, StoryNotFoundError, PartNotFoundError
 from .utils import generate_clean_part_html, slugify
 from .logs import logger
+from .parser import fetch_image