diff --git a/src/lib/waku_message/index.browser.spec.ts b/src/lib/waku_message/index.browser.spec.ts
index c97ea9e4bc..ffb76e6a78 100644
--- a/src/lib/waku_message/index.browser.spec.ts
+++ b/src/lib/waku_message/index.browser.spec.ts
@@ -1,8 +1,9 @@
 import { expect } from 'chai';
 import fc from 'fast-check';
 
-import { WakuMessage } from '../../lib/waku_message';
-import { getPublicKey } from '../../lib/waku_message/version_1';
+import { getPublicKey } from './version_1';
+
+import { WakuMessage } from './index';
 
 describe('Waku Message: Browser & Node', function () {
   it('Waku message round trip binary serialization [clear]', async function () {