diff --git a/checkm8_payloads/CMakeLists.txt b/checkm8_payloads/CMakeLists.txt
index cbf6ad5..ea7005a 100644
--- a/checkm8_payloads/CMakeLists.txt
+++ b/checkm8_payloads/CMakeLists.txt
@@ -5,7 +5,7 @@ set(CMAKE_SYSTEM_PROCESSOR arm)
 set(CMAKE_C_COMPILER /usr/bin/aarch64-linux-gnu-gcc)
 set(CMAKE_ASM_COMPILER /usr/bin/aarch64-linux-gnu-as)
 set(CMAKE_OBJCOPY /usr/bin/aarch64-linux-gnu-objcopy)
-set(CMAKE_C_FLAGS "-nostdlib")
+set(CMAKE_C_FLAGS "-nostdlib -O")
 
 set(PAYLOADS payload_aes)
 add_executable(payload_aes aes.c)
@@ -15,7 +15,7 @@ foreach(BINARY ${PAYLOADS})
     add_custom_command(TARGET ${BINARY} POST_BUILD
                         BYPRODUCTS ${CMAKE_CURRENT_SOURCE_DIR}/bin/${BINARY}.bin
                         COMMAND ${CMAKE_OBJCOPY}
-                        ARGS -O binary -j .text
+                        ARGS -O binary -j .text -j .payload_text
                             ${CMAKE_CURRENT_BINARY_DIR}/${BINARY}
                             ${CMAKE_CURRENT_SOURCE_DIR}/bin/${BINARY}.bin)
 endforeach(BINARY)
\ No newline at end of file
diff --git a/checkm8_payloads/aes.c b/checkm8_payloads/aes.c
index d9d4b17..0f7e011 100644
--- a/checkm8_payloads/aes.c
+++ b/checkm8_payloads/aes.c
@@ -2,6 +2,7 @@
 #include "brfunc_timing.h"
 #include "brfunc_sep.h"
 
+__attribute__ ((section (".payload_text")))
 int aes_hw_crypto_command(unsigned int cmd,
                           void *src,
                           void *dst,