diff --git a/docs/diagrams/sequence/Architecture_Encode_Data.puml b/docs/diagrams/sequence/Architecture_Encode_Data.puml
index 1b6d8edf3f..acbb463d61 100644
--- a/docs/diagrams/sequence/Architecture_Encode_Data.puml
+++ b/docs/diagrams/sequence/Architecture_Encode_Data.puml
@@ -13,6 +13,7 @@ Actor user
 user -> Ui: 'bye'
 Activate Ui
 Ui-> CafeCtrl : receiveUserInput('bye')
+loop !exitCommand
 Activate CafeCtrl
 CafeCtrl -> Parser : parseCommand('bye')
 Activate Parser
@@ -20,6 +21,7 @@ return exitCommand: Command
 CafeCtrl -> Command : exitCommand.execute()
 Activate Command
 return
+end loop
 CafeCtrl->Storage : saveAll()
 Activate Storage
 Storage -> Storage: saveMenu()
diff --git a/docs/images/sequence/Architecture_Encode_Data.png b/docs/images/sequence/Architecture_Encode_Data.png
index 7fb142f6e7..d9aa42dcf2 100644
Binary files a/docs/images/sequence/Architecture_Encode_Data.png and b/docs/images/sequence/Architecture_Encode_Data.png differ