diff --git a/bl1/bl1_fwu.c b/bl1/bl1_fwu.c
index 205ea92..7dd7653 100644
--- a/bl1/bl1_fwu.c
+++ b/bl1/bl1_fwu.c
@@ -52,8 +52,6 @@
 /* Authentication status of each image. */
 extern unsigned int auth_img_flags[];
 
-void cm_set_next_context(void *cpu_context);
-
 /*******************************************************************************
  * Top level handler for servicing FWU SMCs.
  ******************************************************************************/
diff --git a/include/lib/el3_runtime/context_mgmt.h b/include/lib/el3_runtime/context_mgmt.h
index 9479869..eb7a953 100644
--- a/include/lib/el3_runtime/context_mgmt.h
+++ b/include/lib/el3_runtime/context_mgmt.h
@@ -86,6 +86,7 @@
 
 #else
 void *cm_get_next_context(void);
+void cm_set_next_context(void *context);
 #endif /* AARCH32 */
 
 #endif /* __CM_H__ */