diff --git a/libloki/test/node/local_loki_server_test.js b/libloki/test/node/local_loki_server_test.js
index 491b5931c..ed34b7b5f 100644
--- a/libloki/test/node/local_loki_server_test.js
+++ b/libloki/test/node/local_loki_server_test.js
@@ -64,8 +64,10 @@ describe('LocalLokiServer', () => {
       };
 
       const promise = new Promise(res => {
-        server.on('message', message => {
+        server.on('message', eventData => {
+          const { message, onSuccess } = eventData;
           assert.equal(message, 'This is data');
+          onSuccess();
           server.close();
           res();
         });