diff --git a/kasmweb/tests/test.rfb.js b/kasmweb/tests/test.rfb.js
index 4d8c34d..89e45e6 100644
--- a/kasmweb/tests/test.rfb.js
+++ b/kasmweb/tests/test.rfb.js
@@ -107,10 +107,12 @@ describe('Remote Frame Buffer Protocol Client', function () {
             this._rQ = rQ;
         };
 
+        // Avoiding printing the entire Websock buffer on errors
+        Websock.prototype.toString = function () { return "[object Websock]"; };
     });
 
     after(function () {
-        Websock.prototype._allocateBuffers = Websock.prototype._oldAllocateBuffers;
+        delete Websock.prototype.toString;
         this.clock.restore();
         window.requestAnimationFrame = raf;
     });