diff --git a/tests/flytekit/unit/core/test_container_task.py b/tests/flytekit/unit/core/test_container_task.py
index 315e84a13e..6e242681e6 100644
--- a/tests/flytekit/unit/core/test_container_task.py
+++ b/tests/flytekit/unit/core/test_container_task.py
@@ -23,37 +23,33 @@
 
 def test_local_execution():
     try:
-        import docker
-
-        docker.from_env()
+        calculate_ellipse_area_python = ContainerTask(
+            name="calculate_ellipse_area_python",
+            input_data_dir="/var/inputs",
+            output_data_dir="/var/outputs",
+            inputs=kwtypes(a=float, b=float),
+            outputs=kwtypes(area=float, metadata=str),
+            image="ghcr.io/flyteorg/rawcontainers-python:v2",
+            command=[
+                "python",
+                "calculate-ellipse-area.py",
+                "{{.inputs.a}}",
+                "{{.inputs.b}}",
+                "/var/outputs",
+            ],
+        )
+
+        def wf() -> Tuple[float, str]:
+            return calculate_ellipse_area_python(a=3.0, b=4.0)
+
+        area, metadata = wf()
+        assert area == 37.69911184307752
+        assert metadata == "[from python rawcontainer]"
     except Exception as e:
         # Currently, Ubuntu will pass the test, but MacOS and Windows will not
         print(f"Skipping test due to Docker environment setup failure: {e}")
         return
 
-    calculate_ellipse_area_python = ContainerTask(
-        name="calculate_ellipse_area_python",
-        input_data_dir="/var/inputs",
-        output_data_dir="/var/outputs",
-        inputs=kwtypes(a=float, b=float),
-        outputs=kwtypes(area=float, metadata=str),
-        image="ghcr.io/flyteorg/rawcontainers-python:v2",
-        command=[
-            "python",
-            "calculate-ellipse-area.py",
-            "{{.inputs.a}}",
-            "{{.inputs.b}}",
-            "/var/outputs",
-        ],
-    )
-
-    def wf() -> Tuple[float, str]:
-        return calculate_ellipse_area_python(a=3.0, b=4.0)
-
-    area, metadata = wf()
-    assert area == 37.69911184307752
-    assert metadata == "[from python rawcontainer]"
-
 
 def test_pod_template():
     ps = V1PodSpec(