diff --git a/bootloader/hos/hos.c b/bootloader/hos/hos.c
index 1eb3ba85..57cb2f32 100644
--- a/bootloader/hos/hos.c
+++ b/bootloader/hos/hos.c
@@ -214,7 +214,7 @@ static void _hos_eks_save()
 
 		// Get keys.
 		u8 *keys = (u8 *)zalloc(SZ_8K);
-		se_get_aes_keys(keys + SZ_4K, keys, SE_KEY_128_SIZE);
+		se_aes_ctx_get_keys(keys + SZ_4K, keys, SE_KEY_128_SIZE);
 
 		// Set magic and personalized info.
 		h_cfg.eks->magic   = HOS_EKS_MAGIC;
diff --git a/nyx/nyx_gui/hos/hos.c b/nyx/nyx_gui/hos/hos.c
index 07d5b73d..c356c3e3 100644
--- a/nyx/nyx_gui/hos/hos.c
+++ b/nyx/nyx_gui/hos/hos.c
@@ -255,7 +255,7 @@ static void _hos_eks_save()
 
 		// Get keys.
 		u8 *keys = (u8 *)zalloc(SZ_8K);
-		se_get_aes_keys(keys + SZ_4K, keys, SE_KEY_128_SIZE);
+		se_aes_ctx_get_keys(keys + SZ_4K, keys, SE_KEY_128_SIZE);
 
 		// Set magic and personalized info.
 		h_cfg.eks->magic   = HOS_EKS_MAGIC;