diff --git a/flytekit/core/array_node_map_task.py b/flytekit/core/array_node_map_task.py
index aafbaea3ad..5dc0facf36 100644
--- a/flytekit/core/array_node_map_task.py
+++ b/flytekit/core/array_node_map_task.py
@@ -4,9 +4,7 @@
 import logging
 import os  # TODO: use flytekit logger
 from contextlib import contextmanager
-from typing import Dict, List, Optional, Set, Union, cast
-
-from typing_extensions import Any
+from typing import Any, Dict, List, Optional, Set, Union, cast
 
 from flytekit.configuration import SerializationSettings
 from flytekit.core import tracker