diff --git a/htdocs/install/mysql/migration/9.0.0-10.0.0.sql b/htdocs/install/mysql/migration/9.0.0-10.0.0.sql
index 2e143209d9a..66839713bb7 100644
--- a/htdocs/install/mysql/migration/9.0.0-10.0.0.sql
+++ b/htdocs/install/mysql/migration/9.0.0-10.0.0.sql
@@ -107,9 +107,11 @@ ALTER TABLE llx_prelevement_facture_demande ADD COLUMN sourcetype varchar(32);
 ALTER TABLE llx_prelevement_facture_demande ADD COLUMN ext_payment_id varchar(128) NULL;
 ALTER TABLE llx_prelevement_facture_demande ADD COLUMN ext_payment_site varchar(128) NULL;
 
--- Fix if table exists
-ALTER TABLE llx_c_shipment_mode ADD COLUMN entity integer;
+ALTER TABLE llx_c_shipment_mode ADD COLUMN entity integer DEFAULT 1;
+ALTER TABLE llx_c_shipment_mode DROP INDEX uk_c_shipment_mode;
+ALTER TABLE llx_c_shipment_mode ADD UNIQUE INDEX uk_c_shipment_mode (entity, code);
 
+-- Fix if table exists
 ALTER TABLE llx_c_units DROP INDEX uk_c_units_code;
 ALTER TABLE llx_c_units ADD COLUMN scale integer;
 ALTER TABLE llx_c_units ADD COLUMN unit_type varchar(10);