diff --git a/addons/website_crm_partner_assign/controllers/main.py b/addons/website_crm_partner_assign/controllers/main.py
index 56cf9c3a4a0..50ad1e26519 100644
--- a/addons/website_crm_partner_assign/controllers/main.py
+++ b/addons/website_crm_partner_assign/controllers/main.py
@@ -96,6 +96,5 @@ class WebsiteCrmPartnerAssign(http.Controller):
         values = website_partner.get_partner_template_value(partner)
         if not values:
             return self.partners(**post)
-        else:
-            values['main_object'] = values['partner']
+        values['main_object'] = values['partner']
         return request.website.render("website_crm_partner_assign.partner", values)
diff --git a/addons/website_membership/controllers/main.py b/addons/website_membership/controllers/main.py
index feab461f6c3..6985d4449f2 100644
--- a/addons/website_membership/controllers/main.py
+++ b/addons/website_membership/controllers/main.py
@@ -87,4 +87,5 @@ class WebsiteMembership(http.Controller):
         values = website_partner.get_partner_template_value(partner)
         if not values:
             return self.members(**post)
+        values['main_object'] = values['partner']
         return request.website.render("website_membership.partner", values)