diff --git a/components/aws_iot/coremqtt_agent/integration/inc/mqtt_agent_task.h b/components/aws_iot/coremqtt_agent/integration/inc/mqtt_agent_task.h
index 5826e914..0cb71855 100644
--- a/components/aws_iot/coremqtt_agent/integration/inc/mqtt_agent_task.h
+++ b/components/aws_iot/coremqtt_agent/integration/inc/mqtt_agent_task.h
@@ -1,7 +1,7 @@
 /*
  * FreeRTOS V202012.00
  * Copyright (C) 2020 Amazon.com, Inc. or its affiliates.  All Rights Reserved.
- * Copyright 2024 Arm Limited and/or its affiliates
+ * Copyright 2023-2024 Arm Limited and/or its affiliates
  * <open-source-office@arm.com>
  *
  * Permission is hereby granted, free of charge, to any person obtaining a copy of
@@ -56,9 +56,6 @@ struct MQTTAgentCommandContext
     void * pArgs;
 };
 
-void vWaitUntilMQTTAgentReady( void );
-void vWaitUntilMQTTAgentConnected( void );
-bool xIsMqttAgentConnected( void );
 void vStartMqttAgentTask( void );
 
 #endif /* MQTT_AGENT_H */
diff --git a/components/aws_iot/ota_for_aws_iot_embedded_sdk/integration/CMakeLists.txt b/components/aws_iot/ota_for_aws_iot_embedded_sdk/integration/CMakeLists.txt
index 160bdd3b..96b70d2b 100644
--- a/components/aws_iot/ota_for_aws_iot_embedded_sdk/integration/CMakeLists.txt
+++ b/components/aws_iot/ota_for_aws_iot_embedded_sdk/integration/CMakeLists.txt
@@ -27,6 +27,7 @@ target_link_libraries(ota-for-aws-iot-embedded-sdk
         coremqtt
         coremqtt-agent
         freertos-ota-pal-psa
+        helpers-events
         helpers-logging
         tinycbor
 )
diff --git a/components/aws_iot/ota_for_aws_iot_embedded_sdk/integration/src/ota_agent_task.c b/components/aws_iot/ota_for_aws_iot_embedded_sdk/integration/src/ota_agent_task.c
index a38cf3f0..a9a845d4 100644
--- a/components/aws_iot/ota_for_aws_iot_embedded_sdk/integration/src/ota_agent_task.c
+++ b/components/aws_iot/ota_for_aws_iot_embedded_sdk/integration/src/ota_agent_task.c
@@ -35,6 +35,7 @@
 #include "app_config.h"
 
 #include "mqtt_agent_task.h"
+#include "events.h"
 
 /* includes for TFM */
 #include "psa/update.h"