diff --git a/src/database/schemas/models.js b/src/database/schemas/models.js
index 98c37bf..215fd7f 100644
--- a/src/database/schemas/models.js
+++ b/src/database/schemas/models.js
@@ -24,8 +24,10 @@ import { documentJobModel } from './management/documentjob.schema.js';
 import { fileModel } from './management/file.schema.js';
 import { courierServiceModel } from './management/courierservice.schema.js';
 import { courierModel } from './management/courier.schema.js';
-import { taxRateModel } from './management/taxrates.schema.js';
+import { taxRateModel } from './management/taxrate.schema.js';
 import { taxRecordModel } from './management/taxrecord.schema.js';
+import { shipmentModel } from './inventory/shipment.schema.js';
+import { invoiceModel } from './finance/invoice.schema.js';
 
 // Map prefixes to models and id fields
 export const models = {
@@ -98,4 +100,6 @@ export const models = {
   COR: { model: courierModel, idField: '_id', type: 'courier', referenceField: '_reference' },
   TXR: { model: taxRateModel, idField: '_id', type: 'taxRate', referenceField: '_reference' },
   TXD: { model: taxRecordModel, idField: '_id', type: 'taxRecord', referenceField: '_reference' },
+  SHP: { model: shipmentModel, idField: '_id', type: 'shipment', referenceField: '_reference' },
+  INV: { model: invoiceModel, idField: '_id', type: 'invoice', referenceField: '_reference' },
 };