Fix which assignment is being changed. Fixes #580

mozilla · Jun 14, 2017 · dfd420d · dfd420d
1 parent d2b4d97
commit dfd420d
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/webextension/background.js b/webextension/background.js
@@ -448,9 +448,10 @@ const messageHandler = {
         response = assignManager._getByContainer(m.message.userContextId);
         break;
       case "setOrRemoveAssignment":
+        // m.tabId is used for where to place the in content message
+        // m.url is the assignment to be removed/added
         response = browser.tabs.get(m.tabId).then((tab) => {
-          const userContextId = assignManager.getUserContextIdFromCookieStore(tab);
-          return assignManager._setOrRemoveAssignment(tab.id, tab.url, userContextId, m.value);
+          return assignManager._setOrRemoveAssignment(tab.id, m.url, m.userContextId, m.value);
         });
         break;
       case "exemptContainerAssignment":