diff --git a/test/noise.test.ts b/test/noise.test.ts
index 1fceb50..a22dd0b 100644
--- a/test/noise.test.ts
+++ b/test/noise.test.ts
@@ -1,18 +1,13 @@
-import { expect, assert } from "chai";
+import {assert, expect} from "chai";
 import DuplexPair from 'it-pair/duplex';
-import { Noise } from "../src";
+import {Noise} from "../src";
 import {createPeerIdsFromFixtures} from "./fixtures/peer";
 import Wrap from "it-pb-rpc";
-import { random } from "bcrypto";
+import {random} from "bcrypto";
 import sinon from "sinon";
 import {XXHandshake} from "../src/handshake-xx";
-import {
-  createHandshakePayload,
-  generateKeypair,
-  getHandshakePayload, getPayload,
-  signPayload
-} from "../src/utils";
-import {decode0, decode1, encode1, uint16BEDecode, uint16BEEncode} from "../src/encoder";
+import {createHandshakePayload, generateKeypair, getHandshakePayload, getPayload, signPayload} from "../src/utils";
+import {decode0, decode2, encode1, uint16BEDecode, uint16BEEncode} from "../src/encoder";
 import {XX} from "../src/handshakes/xx";
 import {Buffer} from "buffer";
 import {getKeyPairFromPeerId} from "./utils";
@@ -89,7 +84,7 @@ describe("Noise", () => {
         wrapped.writeLP(encode1(messageBuffer));
 
         // Stage 2 - finish handshake
-        receivedMessageBuffer = decode1((await wrapped.readLP()).slice());
+        receivedMessageBuffer = decode2((await wrapped.readLP()).slice());
         xx.recvMessage(handshake.session, receivedMessageBuffer);
         return {wrapped, handshake};
       })(),