diff --git a/cmake/install.cmake b/cmake/install.cmake
index ab6292a..f13d3bf 100644
--- a/cmake/install.cmake
+++ b/cmake/install.cmake
@@ -111,6 +111,13 @@
             DESTINATION ${INSTALL_INTERFACE_INC_DIR})
 endif()
 
+if(TFM_PARTITION_FIRMWARE_UPDATE)
+    install(FILES       ${INTERFACE_INC_DIR}/psa/update.h
+            DESTINATION ${INSTALL_INTERFACE_INC_DIR}/psa)
+    install(FILES       ${INTERFACE_INC_DIR}/tfm_fwu_defs.h
+            DESTINATION ${INSTALL_INTERFACE_INC_DIR})
+endif()
+
 ####################### export sources #########################################
 
 if (TFM_MULTI_CORE_TOPOLOGY)
@@ -220,3 +227,10 @@
                 DESTINATION ${INSTALL_IMAGE_SIGNING_DIR}/keys)
     endif()
 endif()
+
+if(TFM_PARTITION_FIRMWARE_UPDATE)
+    if(NOT TFM_PSA_API)
+        install(FILES       ${INTERFACE_SRC_DIR}/tfm_firmware_update_func_api.c
+                DESTINATION ${INSTALL_INTERFACE_SRC_DIR})
+    endif()
+endif()