diff --git a/spec/migrations/20230726144458_swap_notes_id_to_bigint_for_self_managed_spec.rb b/spec/migrations/20230726144458_swap_notes_id_to_bigint_for_self_managed_spec.rb
index bef9452f39718881df2e088b6c521a4d479bd1fe..96151de1148015dffbbb86bfd0d5b2576131bc15 100644
--- a/spec/migrations/20230726144458_swap_notes_id_to_bigint_for_self_managed_spec.rb
+++ b/spec/migrations/20230726144458_swap_notes_id_to_bigint_for_self_managed_spec.rb
@@ -126,6 +126,9 @@
 
         after do
           connection.execute('ALTER TABLE system_note_metadata DROP CONSTRAINT IF EXISTS fk_rails_d83a918cb1')
+          connection.execute('ALTER TABLE system_note_metadata DROP CONSTRAINT IF EXISTS fk_d83a918cb1_tmp')
+          connection.execute('ALTER TABLE system_note_metadata ADD CONSTRAINT fk_d83a918cb1 FOREIGN KEY (note_id)
+          REFERENCES notes(id) ON DELETE CASCADE')
         end
 
         it 'swaps the columns' do