diff --git a/lib/nodejs/test/client.js b/lib/nodejs/test/client.js
index 49e3a5e..31ea06e 100644
--- a/lib/nodejs/test/client.js
+++ b/lib/nodejs/test/client.js
@@ -69,6 +69,11 @@
   type = "http";
 }
 
+if (program.transport === "websocket") {
+  program.transport = "buffered";
+  type = "websocket";
+}
+
 const options = {
   transport: helpers.transports[program.transport],
   protocol: helpers.protocols[program.protocol]
diff --git a/lib/nodejs/test/server.js b/lib/nodejs/test/server.js
index 7402094..677839a 100644
--- a/lib/nodejs/test/server.js
+++ b/lib/nodejs/test/server.js
@@ -61,6 +61,9 @@
 if (program.transport === "http") {
   program.transport = "buffered";
   type = "http";
+} else if (program.transport === "websocket") {
+  program.transport = "buffered";
+  type = "websocket";
 }
 
 let options = {