diff --git a/providers/amazon/src/airflow/providers/amazon/aws/sensors/s3.py b/providers/amazon/src/airflow/providers/amazon/aws/sensors/s3.py
index c094b0d5ccfd9..42cdbd402ee19 100644
--- a/providers/amazon/src/airflow/providers/amazon/aws/sensors/s3.py
+++ b/providers/amazon/src/airflow/providers/amazon/aws/sensors/s3.py
@@ -36,7 +36,7 @@
 from airflow.providers.amazon.aws.sensors.base_aws import AwsBaseSensor
 from airflow.providers.amazon.aws.triggers.s3 import S3KeysUnchangedTrigger, S3KeyTrigger
 from airflow.providers.amazon.aws.utils.mixins import aws_template_fields
-from airflow.sensors.base import poke_mode_only
+from airflow.providers.amazon.version_compat import poke_mode_only
 
 
 class S3KeySensor(AwsBaseSensor[S3Hook]):
diff --git a/providers/amazon/src/airflow/providers/amazon/version_compat.py b/providers/amazon/src/airflow/providers/amazon/version_compat.py
index f2e3fdfc6afc5..87cdd8c94d953 100644
--- a/providers/amazon/src/airflow/providers/amazon/version_compat.py
+++ b/providers/amazon/src/airflow/providers/amazon/version_compat.py
@@ -37,8 +37,10 @@ def get_base_airflow_version_tuple() -> tuple[int, int, int]:
 
 if AIRFLOW_V_3_1_PLUS:
     from airflow.sdk import BaseHook
+    from airflow.sdk.bases.sensor import poke_mode_only
 else:
     from airflow.hooks.base import BaseHook  # type: ignore[attr-defined,no-redef]
+    from airflow.sensors.base import poke_mode_only  # type: ignore[no-redef]
 
 if AIRFLOW_V_3_0_PLUS:
     from airflow.sdk import BaseOperator, BaseOperatorLink, BaseSensorOperator
@@ -55,5 +57,6 @@ def get_base_airflow_version_tuple() -> tuple[int, int, int]:
     "BaseOperator",
     "BaseOperatorLink",
     "BaseSensorOperator",
+    "poke_mode_only",
     "XCom",
 ]