diff --git a/InventoryDistanceBasedSourceSelection/etc/db_schema.xml b/InventoryDistanceBasedSourceSelection/etc/db_schema.xml
index 0f30b30bd8d3..dd5582c2c234 100644
--- a/InventoryDistanceBasedSourceSelection/etc/db_schema.xml
+++ b/InventoryDistanceBasedSourceSelection/etc/db_schema.xml
@@ -8,6 +8,8 @@
 <schema xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
         xsi:noNamespaceSchemaLocation="urn:magento:framework:Setup/Declaration/Schema/etc/schema.xsd">
     <table name="inventory_geoname" resource="default" engine="innodb">
+        <column xsi:type="int" name="entity_id" unsigned="true" nullable="false" identity="true"
+                comment="entity id"/>
         <column xsi:type="varchar" name="country_code" nullable="false" length="64"/>
         <column xsi:type="varchar" name="postcode" nullable="false" length="64"/>
         <column xsi:type="varchar" name="city" nullable="false" length="180"/>
@@ -15,5 +17,8 @@
         <column xsi:type="varchar" name="province" nullable="false" length="64"/>
         <column xsi:type="double" name="latitude" nullable="false" />
         <column xsi:type="double" name="longitude" nullable="false"/>
+        <constraint xsi:type="primary" referenceId="PRIMARY">
+            <column name="entity_id"/>
+        </constraint>
     </table>
 </schema>
diff --git a/InventoryDistanceBasedSourceSelection/etc/db_schema_whitelist.json b/InventoryDistanceBasedSourceSelection/etc/db_schema_whitelist.json
index f6abc16afada..63411e22d5b1 100644
--- a/InventoryDistanceBasedSourceSelection/etc/db_schema_whitelist.json
+++ b/InventoryDistanceBasedSourceSelection/etc/db_schema_whitelist.json
@@ -1,6 +1,7 @@
 {
     "inventory_geoname": {
         "column": {
+            "entity_id": true,
             "country_code": true,
             "zip_code": true,
             "city": true,
@@ -9,6 +10,9 @@
             "latitude": true,
             "longitude": true,
             "postcode": true
+        },
+        "constraint": {
+            "PRIMARY": true
         }
     }
 }
\ No newline at end of file