diff --git a/packages/dock-blockchain-modules/tests/integration/modules/attest-module.test.js b/packages/dock-blockchain-modules/tests/integration/modules/attest-module.test.js
index 28a185182..85b908695 100644
--- a/packages/dock-blockchain-modules/tests/integration/modules/attest-module.test.js
+++ b/packages/dock-blockchain-modules/tests/integration/modules/attest-module.test.js
@@ -27,6 +27,6 @@ describe("AttestModule", () => {
 
   generateAttestModuleTests(
     { did: new DockDIDModule(dock), attest: new DockAttestModule(dock) },
-    DockDid
+    { DID: DockDid }
   );
 });
diff --git a/packages/dock-blockchain-modules/tests/integration/modules/did-module.test.js b/packages/dock-blockchain-modules/tests/integration/modules/did-module.test.js
index 95383099c..4eb35f615 100644
--- a/packages/dock-blockchain-modules/tests/integration/modules/did-module.test.js
+++ b/packages/dock-blockchain-modules/tests/integration/modules/did-module.test.js
@@ -26,7 +26,7 @@ describe("DIDModule", () => {
 
   didModuleTests(
     { did: new DockDIDModule(dock) },
-    DockDid,
+    { DID: DockDid },
     (name) =>
       name !== "Creates `DIDDocument` containing BBS/BBSPlus/PS keys" &&
       name !== "Creates `DIDDocument` containing services"