diff --git a/tests/test.rfb.js b/tests/test.rfb.js
index edb98cb4..432bcba9 100644
--- a/tests/test.rfb.js
+++ b/tests/test.rfb.js
@@ -115,6 +115,7 @@ describe('Remote Frame Buffer Protocol Client', function () {
     });
 
     after(function () {
+        Websock.prototype._allocateBuffers = Websock.prototype._oldAllocateBuffers;
         delete Websock.prototype.inspect;
         this.clock.restore();
         window.requestAnimationFrame = raf;