diff --git a/lib/websocket.js b/lib/websocket.js
index 5b40075e4..6d61d35d6 100644
--- a/lib/websocket.js
+++ b/lib/websocket.js
@@ -71,7 +71,7 @@ WebSocketProvider.prototype.unload = function() {
     this.ws.close();
 };
 Object.defineProperty(WebSocketProvider.prototype, "onmessage", {
-    set: function(provider) { this.onMessage(provider); }
+    set: function(handler) { this.onMessage(handler); }
 });
 
 if (typeof(module) !== "undefined")