From dfd420d1a5cc064dd898295e9e4403d310dd928b Mon Sep 17 00:00:00 2001
From: Jonathan Kingston <jonathan@jooped.co.uk>
Date: Wed, 14 Jun 2017 15:10:05 +0100
Subject: [PATCH] Fix which assignment is being changed. Fixes #580

---
 webextension/background.js | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/webextension/background.js b/webextension/background.js
index 9d64504..7b43b3c 100644
--- 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":