diff --git a/mass_mailing_custom_unsubscribe/tests/test_ui.py b/mass_mailing_custom_unsubscribe/tests/test_ui.py
index 9f8e24b7..07647b73 100644
--- a/mass_mailing_custom_unsubscribe/tests/test_ui.py
+++ b/mass_mailing_custom_unsubscribe/tests/test_ui.py
@@ -3,6 +3,7 @@
 import mock
 from contextlib import contextmanager
 from odoo.tests.common import HttpCase
+from werkzeug import urls
 
 
 class UICase(HttpCase):
@@ -11,7 +12,7 @@ class UICase(HttpCase):
 
     def extract_url(self, mail, *args, **kwargs):
         url = mail._get_unsubscribe_url(self.email)
-        self.assertIn("&token=", url)
+        self.assertTrue(urls.url_parse(url).decode_query().get('token'))
         self.assertTrue(url.startswith(self.domain))
         self.url = url.replace(self.domain, "", 1)
         return True