diff --git a/src/codec.ts b/src/codec.ts
index 84e6a12..ae7ba15 100644
--- a/src/codec.ts
+++ b/src/codec.ts
@@ -1,6 +1,12 @@
 import debug from "debug";
-import { DecodedMessage } from "@waku/core/lib/message/version_0";
-import { IEncoder, IDecoder, IDecodedMessage, IProtoMessage, IMessage } from "@waku/interfaces";
+import { DecodedMessage } from "@waku/core";
+import {
+  IEncoder,
+  IDecoder,
+  IDecodedMessage,
+  IProtoMessage,
+  IMessage,
+} from "@waku/interfaces";
 import { WakuMessage } from "@waku/proto";
 
 import { HandshakeResult, HandshakeStepResult } from "./handshake.js";