diff --git a/core/service/storage/container.py b/core/service/storage/container.py
index 97b9f44c88..ad9d879dd5 100644
--- a/core/service/storage/container.py
+++ b/core/service/storage/container.py
@@ -1,13 +1,15 @@
-from typing import Optional
+from typing import TYPE_CHECKING, Optional
 
 import boto3
 from dependency_injector import providers
 from dependency_injector.containers import DeclarativeContainer
 from dependency_injector.providers import Provider, Singleton
-from mypy_boto3_s3 import S3Client
 
 from core.service.storage.s3 import S3Service
 
+if TYPE_CHECKING:
+    from mypy_boto3_s3 import S3Client
+
 
 class Storage(DeclarativeContainer):
     config = providers.Configuration()