diff --git a/easyllm/__init__.py b/easyllm/__init__.py
index e4e25bf..6ea7dcc 100644
--- a/easyllm/__init__.py
+++ b/easyllm/__init__.py
@@ -1,4 +1,4 @@
 # SPDX-FileCopyrightText: 2023-present philschmid <schmidphilipp1995@gmail.com>
 #
 # SPDX-License-Identifier: MIT
-__version__ = "0.7.0.dev0"
+__version__ = "0.6.1"
diff --git a/easyllm/utils/aws.py b/easyllm/utils/aws.py
index b491130..e69cf7c 100644
--- a/easyllm/utils/aws.py
+++ b/easyllm/utils/aws.py
@@ -7,7 +7,6 @@
 from datetime import datetime
 from typing import Optional
 
-from botocore.config import Config
 from requests import __version__ as requests_version
 from requests.auth import AuthBase
 from requests.compat import urlparse
@@ -15,6 +14,7 @@
 
 try:
     import boto3
+    from botocore.config import Config
 except ImportError:
     boto3 = None