diff --git a/agent/lib_aws_sdk_php.h b/agent/lib_aws_sdk_php.h
index 1b52a4772..2e6bfc74c 100644
--- a/agent/lib_aws_sdk_php.h
+++ b/agent/lib_aws_sdk_php.h
@@ -7,6 +7,8 @@
 #ifndef LIB_AWS_SDK_PHP_HDR
 #define LIB_AWS_SDK_PHP_HDR
 
+#include "nr_segment_message.h"
+
 #if ZEND_MODULE_API_NO >= ZEND_8_1_X_API_NO /* PHP8.1+ */
 /* Service instrumentation only supported above PHP 8.1+*/
 
diff --git a/agent/php_mshutdown.c b/agent/php_mshutdown.c
index f3ac15b16..189960240 100644
--- a/agent/php_mshutdown.c
+++ b/agent/php_mshutdown.c
@@ -16,6 +16,7 @@
 #include "nr_agent.h"
 #include "util_logging.h"
 #include "fw_wordpress.h"
+#include "lib_aws_sdk_php.h"
 
 #ifdef TAGS
 void zm_shutdown_newrelic(void); /* ctags landing pad only */