diff --git a/src/ui/menu-ui-handler.ts b/src/ui/menu-ui-handler.ts
index e88db0914..7850b9f51 100644
--- a/src/ui/menu-ui-handler.ts
+++ b/src/ui/menu-ui-handler.ts
@@ -277,8 +277,10 @@ export default class MenuUiHandler extends MessageUiHandler {
       }
     } else if (button === Button.CANCEL) {
       success = true;
-      if (!ui.revertMode())
-        ui.setMode(Mode.MESSAGE);
+      ui.revertMode().then(result => {
+        if (!result)
+           ui.setMode(Mode.MESSAGE);
+      });
     } else {
       switch (button) {
         case Button.UP: