diff --git a/src/crewai_tools/tools/code_interpreter_tool/code_interpreter_tool.py b/src/crewai_tools/tools/code_interpreter_tool/code_interpreter_tool.py
index b508e4b6a..2a0f9ffe6 100644
--- a/src/crewai_tools/tools/code_interpreter_tool/code_interpreter_tool.py
+++ b/src/crewai_tools/tools/code_interpreter_tool/code_interpreter_tool.py
@@ -4,6 +4,8 @@ from typing import List, Optional, Type
 
 from crewai.tools import BaseTool
 from docker import from_env as docker_from_env
+from docker import DockerClient
+from docker.models.containers import Container
 from docker.errors import ImageNotFound, NotFound
 from docker.models.containers import Container
 from pydantic import BaseModel, Field
@@ -46,7 +48,7 @@ class CodeInterpreterTool(BaseTool):
         client = (
             docker_from_env()
             if self.user_docker_base_url == None
-            else docker.DockerClient(base_url=self.user_docker_base_url)
+            else DockerClient(base_url=self.user_docker_base_url)
         )
 
         try:
@@ -137,4 +139,4 @@ class CodeInterpreterTool(BaseTool):
             exec(code, {}, exec_locals)
             return exec_locals.get("result", "No result variable found.")
         except Exception as e:
-            return f"An error occurred: {str(e)}"
+            return f"An error occurred: {str(e)}"
\ No newline at end of file