diff --git a/js/ui/endSessionDialog.js b/js/ui/endSessionDialog.js
index c9ddce8db..7606d2471 100644
--- a/js/ui/endSessionDialog.js
+++ b/js/ui/endSessionDialog.js
@@ -116,7 +116,7 @@ const restartInstallDialogContent = {
                         seconds).format(seconds);
     },
     confirmButtons: [{ signal: 'ConfirmedReboot',
-                       label:  C_("button", "Restart & Install") }],
+                       label:  C_("button", "Restart & Install") }],
     iconName: 'view-refresh-symbolic',
     iconStyleClass: 'end-session-dialog-shutdown-icon',
     showOtherSessions: true,
@@ -300,9 +300,6 @@ const EndSessionDialog = new Lang.Class({
         if (!open)
             return;
 
-        if (this._type == 2 && this._updatesFile.query_exists(null))
-            this._type = 3;
-
         let dialogContent = DialogContent[this._type];
 
         let subject = dialogContent.subject;
@@ -545,6 +542,9 @@ const EndSessionDialog = new Lang.Class({
         this._totalSecondsToStayOpen = totalSecondsToStayOpen;
         this._type = type;
 
+        if (this._type == 2 && this._updatesFile.query_exists(null))
+            this._type = 3;
+
         this._applications = [];
         this._applicationList.destroy_all_children();