diff --git a/providers/apache/kafka/src/airflow/providers/apache/kafka/triggers/await_message.py b/providers/apache/kafka/src/airflow/providers/apache/kafka/triggers/await_message.py
index 80d2a418e6fa3..83d3f9b658d3e 100644
--- a/providers/apache/kafka/src/airflow/providers/apache/kafka/triggers/await_message.py
+++ b/providers/apache/kafka/src/airflow/providers/apache/kafka/triggers/await_message.py
@@ -25,11 +25,17 @@
 
 from airflow.exceptions import AirflowException
 from airflow.providers.apache.kafka.hooks.consume import KafkaConsumerHook
-from airflow.triggers.base import BaseTrigger, TriggerEvent
+from airflow.providers.apache.kafka.version_compat import AIRFLOW_V_3_0_PLUS
+from airflow.triggers.base import TriggerEvent
 from airflow.utils.module_loading import import_string
 
+if AIRFLOW_V_3_0_PLUS:
+    from airflow.triggers.base import BaseEventTrigger
+else:
+    from airflow.triggers.base import BaseTrigger as BaseEventTrigger  # type: ignore
 
-class AwaitMessageTrigger(BaseTrigger):
+
+class AwaitMessageTrigger(BaseEventTrigger):
     """
     A trigger that waits for a message matching specific criteria to arrive in Kafka.