From 0040bf6acd42b32ebc0699b62b37261eeca6a4ea Mon Sep 17 00:00:00 2001
From: "Emmanuel .A" <asaoluemma@gmail.com>
Date: Wed, 7 Jan 2026 09:59:42 +0100
Subject: [PATCH] improve buildBaseAddress

---
 packages/mesh-core-cst/src/utils/builder.ts | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/packages/mesh-core-cst/src/utils/builder.ts b/packages/mesh-core-cst/src/utils/builder.ts
index 7df458a8d..f2ced239b 100644
--- a/packages/mesh-core-cst/src/utils/builder.ts
+++ b/packages/mesh-core-cst/src/utils/builder.ts
@@ -29,16 +29,18 @@ export const buildBaseAddress = (
   networkId: number,
   paymentKeyHash: Hash28ByteBase16,
   stakeKeyHash: Hash28ByteBase16,
+  paymentCredType: CredentialType = CredentialType.KeyHash,
+  stakeCredType: CredentialType = CredentialType.KeyHash
 ): BaseAddress => {
   return BaseAddress.fromCredentials(
     networkId,
     {
       hash: paymentKeyHash,
-      type: CredentialType.KeyHash,
+      type: paymentCredType,
     },
     {
       hash: stakeKeyHash,
-      type: CredentialType.KeyHash,
+      type: stakeCredType,
     },
   );
 };