diff --git a/platform/firefox/bootstrap.js b/platform/firefox/bootstrap.js
index b634976a1..fec58275d 100644
--- a/platform/firefox/bootstrap.js
+++ b/platform/firefox/bootstrap.js
@@ -74,7 +74,7 @@ function startup(data, reason) {
         );
     };
 
-    if (reason !== APP_STARTUP) {
+    if ( reason !== APP_STARTUP ) {
         onReady();
         return;
     }
diff --git a/platform/firefox/vapi-background.js b/platform/firefox/vapi-background.js
index ce792d1c3..10cb25deb 100644
--- a/platform/firefox/vapi-background.js
+++ b/platform/firefox/vapi-background.js
@@ -1139,7 +1139,7 @@ var httpObserver = {
         if ( !(channel instanceof Ci.nsIHttpChannel) ) {
             return;
         }
-        
+
         var URI = channel.URI;
         var channelData, result;
 
@@ -1483,7 +1483,7 @@ vAPI.toolbarButton.init = function() {
                 'content: attr(badge);',
             '}'
         );
-     } else {
+    } else {
         this.CUIEvents = {};
 
         var updateBadge = function() {
@@ -1584,7 +1584,6 @@ vAPI.toolbarButton.init = function() {
     }.bind(this));
 };
 
-
 /******************************************************************************/
 
 vAPI.toolbarButton.onBeforeCreated = function(doc) {
@@ -1654,6 +1653,27 @@ vAPI.toolbarButton.onViewHiding = function({target}) {
     target.firstChild.setAttribute('src', 'about:blank');
 };
 
+vAPI.toolbarButton.updateState = function(win, tabId) {
+    var button = win.document.getElementById(this.id);
+
+    if ( !button ) {
+        return;
+    }
+
+    var icon = this.tabs[tabId];
+    button.setAttribute('badge', icon && icon.badge || '');
+
+    if ( !icon || !icon.img ) {
+        icon = '';
+    }
+    else {
+        icon = 'url(' + vAPI.getURL('img/browsericons/icon16.svg') + ')';
+    }
+
+    button.style.listStyleImage = icon;
+};
+}
+
 /******************************************************************************/
 
 vAPI.toolbarButton.init();