diff --git a/src/database/schemas/finance/invoice.schema.js b/src/database/schemas/finance/invoice.schema.js
index 89cb964..ed7e475 100644
--- a/src/database/schemas/finance/invoice.schema.js
+++ b/src/database/schemas/finance/invoice.schema.js
@@ -33,8 +33,10 @@ const invoiceSchema = new Schema(
     shippingAmountWithTax: { type: Number, required: true, default: 0 },
     grandTotalAmount: { type: Number, required: true, default: 0 },
     totalTaxAmount: { type: Number, required: true, default: 0 },
-    from: { type: Schema.Types.ObjectId, ref: 'vendor', required: false },
-    to: { type: Schema.Types.ObjectId, ref: 'client', required: false },
+    from: { type: Schema.Types.ObjectId, refPath: 'fromType', required: false },
+    fromType: { type: String, required: false },
+    to: { type: Schema.Types.ObjectId, refPath: 'toType', required: false },
+    toType: { type: String, required: false },
     state: {
       type: { type: String, required: true, default: 'draft' },
     },