diff --git a/xmlsitemap.install b/xmlsitemap.install
index af3f3fd..2c899c3 100644
--- a/xmlsitemap.install
+++ b/xmlsitemap.install
@@ -571,8 +571,10 @@ function xmlsitemap_update_7203() {
  * Add new primary key including the language.
  */
 function xmlsitemap_update_7204() {
-  db_drop_primary_key('xmlsitemap');
-  db_add_primary_key('xmlsitemap', array('id', 'type', 'language'));
+  $table = 'xmlsitemap';
+  db_drop_primary_key($table);
+  db_drop_unique_key($table, 'language');
+  db_add_primary_key($table, array('id', 'type', 'language'));
 }
 
 /**