diff --git a/addons/website_customer/controllers/main.py b/addons/website_customer/controllers/main.py
index 21aca3b6235..585e7837a74 100644
--- a/addons/website_customer/controllers/main.py
+++ b/addons/website_customer/controllers/main.py
@@ -81,11 +81,11 @@ class WebsiteCustomer(http.Controller):
             return self.customers(**post)
 
         partner_obj = request.registry['res.partner']
-        if values['partner_data']['assigned_partner_id']:
+        if values['partner_data'].get('assigned_partner_id', None):
             values['assigned_partner_data'] = partner_obj.read(
                 request.cr, openerp.SUPERUSER_ID, [values['partner_data']['assigned_partner_id'][0]],
                 request.website.get_partner_white_list_fields(), context=request.context)[0]
-        if values['partner_data']['implemented_partner_ids']:
+        if values['partner_data'].get('implemented_partner_ids', None):
             implemented_partners_data = partner_obj.read(
                 request.cr, openerp.SUPERUSER_ID, values['partner_data']['implemented_partner_ids'],
                 request.website.get_partner_white_list_fields(), context=request.context)