diff --git a/framework/test/unit/modules/random/endpoint.spec.ts b/framework/test/unit/modules/random/endpoint.spec.ts
index 5a61c63f91..a1065abf89 100644
--- a/framework/test/unit/modules/random/endpoint.spec.ts
+++ b/framework/test/unit/modules/random/endpoint.spec.ts
@@ -13,7 +13,7 @@
  */
 
 import * as cryptography from '@liskhq/lisk-cryptography';
-import { ModuleEndpointContext, RandomModule } from '../../../../src';
+import { ModuleEndpointContext, Modules } from '../../../../src';
 import { RandomEndpoint } from '../../../../src/modules/random/endpoint';
 import { HashOnionStore } from '../../../../src/modules/random/stores/hash_onion';
 import {
@@ -77,7 +77,7 @@ describe('RandomModuleEndpoint', () => {
 	};
 
 	beforeEach(async () => {
-		const randomModule = new RandomModule();
+		const randomModule = new Modules.Random.RandomModule();
 		randomEndpoint = new RandomEndpoint(randomModule.stores, randomModule.offchainStores);
 		const stateStore = new PrefixedStateReadWriter(new InMemoryPrefixedStateDB());
 		context = createTransientModuleEndpointContext({
diff --git a/framework/test/unit/modules/token/endpoint.spec.ts b/framework/test/unit/modules/token/endpoint.spec.ts
index b25cbd662e..3dc5d29357 100644
--- a/framework/test/unit/modules/token/endpoint.spec.ts
+++ b/framework/test/unit/modules/token/endpoint.spec.ts
@@ -32,7 +32,7 @@ import { ModuleConfig } from '../../../../src/modules/token/types';
 import { InternalMethod } from '../../../../src/modules/token/internal_method';
 
 describe('token endpoint', () => {
-	const tokenModule = new TokenModule();
+	const tokenModule = new Modules.Token.TokenModule();
 	const addr = utils.getRandomBytes(20);
 	const mainChainID = Buffer.from([1, 0, 0, 0]);
 	const mainChainTokenID = Buffer.concat([mainChainID, Buffer.from([0, 0, 0, 0])]);