diff --git a/xmlsitemap.install b/xmlsitemap.install
index 916ebec..f89a030 100644
--- a/xmlsitemap.install
+++ b/xmlsitemap.install
@@ -239,7 +239,7 @@ function xmlsitemap_schema() {
         'default' => 0,
       ),
     ),
-    'primary key' => array('id', 'type'),
+    'primary key' => array('id', 'type', 'language'),
     'indexes' => array(
       'loc' => array('loc'),
       'access_status_loc' => array('access', 'status', 'loc'),
@@ -568,3 +568,11 @@ function _xmlsitemap_sitemap_rehash_all() {
     }
   }
 }
+
+/**
+ * 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'));
+}
diff --git a/xmlsitemap_node/xmlsitemap_node.install b/xmlsitemap_node/xmlsitemap_node.install
index 110c411..c7eaa5b 100644
--- a/xmlsitemap_node/xmlsitemap_node.install
+++ b/xmlsitemap_node/xmlsitemap_node.install
@@ -42,11 +42,3 @@ function xmlsitemap_node_update_6200() {
  */
 function xmlsitemap_node_update_6201() {
 }
-
-/**
- * Empty update.
- */
-function xmlsitemap_node_update_6202() {
-  db_query('ALTER TABLE {xmlsitemap} DROP PRIMARY KEY');
-  db_add_primary_key('xmlsitemap', array('id', 'type', 'language'));
-}