1 files changed, 3 insertions, 1 deletions

diff --git a/hosts/surtr/postgresql/default.nix b/hosts/surtr/postgresql/default.nix
index 79b17f5a..5969ee18 100644
--- a/hosts/surtr/postgresql/default.nix
+++ b/hosts/surtr/postgresql/default.nix
@@ -226,10 +226,12 @@ in {
           ALTER TABLE mailbox_mapping DROP CONSTRAINT mailbox_mapping_extension_check;
           ALTER TABLE mailbox_mapping ADD CONSTRAINT mailbox_mapping_extension_check CHECK (CASE WHEN extension IS NOT NULL THEN extension NOT LIKE '%+%' AND extension <> ''' AND local IS DISTINCT FROM ''' ELSE true END);
 
-          DROP INDEX local_domain_unique;
+          ALTER TABLE mailbox_mapping DROP CONSTRAINT local_domain_unique;
           CREATE UNIQUE INDEX local_domain_unique ON (local, domain) WHERE extension IS null;
+          ALTER TABLE mailbox_mapping ADD CONSTRAINT local_domain_unique USING INDEX local_domain_unique;
 
           CREATE UNIQUE INDEX local_extension_domain_unique ON (local, extension, domain);
+          ALTER TABLE mailbox_mapping ADD CONSTRAINT local_extension_domain_unique USING INDEX local_extension_domain_unique;
 
           ALTER TABLE mailbox_mapping ADD COLUMN reject bool NOT NULL DEFAULT false;