diff --git a/js/ui/endSessionDialog.js b/js/ui/endSessionDialog.js
index dc3ad81a2..966bae20e 100644
--- a/js/ui/endSessionDialog.js
+++ b/js/ui/endSessionDialog.js
@@ -140,10 +140,10 @@ const DialogType = {
 };
 
 const DialogContent = {
-    DialogType.LOGOUT: logoutDialogContent,
-    DialogType.SHUTDOWN: shutdownDialogContent,
-    DialogType.RESTART: restartDialogContent,
-    DialogType.UPDATE_RESTART: restartInstallDialogContent
+    0 /* DialogType.LOGOUT */: logoutDialogContent,
+    1 /* DialogType.SHUTDOWN */: shutdownDialogContent,
+    2 /* DialogType.RESTART */: restartDialogContent,
+    3 /* DialogType.UPDATE_RESTART */: restartInstallDialogContent
 };
 
 const MAX_USERS_IN_SESSION_DIALOG = 5;