diff --git a/source/uml/bl33_swd_capsule_install.txt b/source/uml/bl33_swd_capsule_install.txt
index bdd89ccdbbf9351e98cfd21268aaa5357b9e1f76..2bd46664a46c6f92386468f4c882163e46afc2ac 100755
--- a/source/uml/bl33_swd_capsule_install.txt
+++ b/source/uml/bl33_swd_capsule_install.txt
@@ -11,5 +11,6 @@ OS -> UEFI ++: call CapsuleUpdate()
 UEFI -> FWU ++: transfer FW blobs
 FWU -> UEFI --
 UEFI -> OS --
-OS -> TFA: PSCI_SYSTEM_RESET
+OS -> UEFI ++: call ResetSystem()
+UEFI -> TFA: PSCI_SYSTEM_RESET
 @enduml
diff --git a/source/uml/bl33_swd_img_test_fail.txt b/source/uml/bl33_swd_img_test_fail.txt
index 3a2c5976bc8250bd19b1c84df9b6b1003216e695..c97c9ad174fafebf19eb4ea1ed3bf2f231a3d2e4 100644
--- a/source/uml/bl33_swd_img_test_fail.txt
+++ b/source/uml/bl33_swd_img_test_fail.txt
@@ -27,6 +27,7 @@ OS -> UEFI ++: install FW revert capsule
 UEFI -> FWU ++: set previously working bank as active
 FWU -> UEFI --
 UEFI -> OS --
-OS -> TFA : PSCI_SYSTEM_RESET
+OS -> UEFI ++: call ResetSystem()
+UEFI -> TFA : PSCI_SYSTEM_RESET
 
 @enduml