diff --git a/docker-compose.yml b/docker-compose.yml
index be0c4af..041cd4a 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -3,7 +3,7 @@ version: "2.1"
 services:
   flaresolverr:
     # DockerHub mirror flaresolverr/flaresolverr:latest
-    image: ghcr.io/flaresolverr/flaresolverr:latest
+    image: flaresolverr-custom:latest
     container_name: flaresolverr
     environment:
       - LOG_LEVEL=${LOG_LEVEL:-info}
diff --git a/src/flaresolverr_service.py b/src/flaresolverr_service.py
index c353fe4..c745f6d 100644
--- a/src/flaresolverr_service.py
+++ b/src/flaresolverr_service.py
@@ -1,5 +1,6 @@
 import json
 import logging
+import traceback
 import platform
 import sys
 import time
@@ -422,10 +423,6 @@ def _evil_logic(req: V1RequestBase, driver: WebDriver, method: str) -> Challenge
 
 
 def _post_request(req: V1RequestBase, driver: WebDriver):
-    import logging
-    import traceback
-    import time
-
     try:
         content_type = getattr(req, 'contentType', 'application/x-www-form-urlencoded')
         headers = getattr(req, 'headers', {})