diff --git a/lib/Client/InternalClient.js b/lib/Client/InternalClient.js
index fe9c623c9..5fd1b4436 100644
--- a/lib/Client/InternalClient.js
+++ b/lib/Client/InternalClient.js
@@ -1780,30 +1780,30 @@ var InternalClient = (function () {
 			self.sendWS(data);
 		};
 
-		this.websocket.onclose = function (code) {
+		this.websocket.onclose = function (event) {
 			self.websocket = null;
 			self.state = _ConnectionState2["default"].DISCONNECTED;
-			if (code) {
-				client.emit("warn", "WS close: " + code);
+			if (event && event.code) {
+				client.emit("warn", "WS close: " + event.code);
 				var err;
-				if (code === 4001) {
+				if (event.code === 4001) {
 					err = new Error("Gateway received invalid OP code");
-				} else if (code === 4005) {
+				} else if (event.code === 4005) {
 					err = new Error("Gateway received invalid message");
-				} else if (code === 4003) {
+				} else if (event.code === 4003) {
 					err = new Error("Not authenticated");
-				} else if (code === 4004) {
+				} else if (event.code === 4004) {
 					err = new Error("Authentication failed");
-				} else if (code === 4005) {
+				} else if (event.code === 4005) {
 					err = new Error("Already authenticated");
-				}if (code === 4006 || code === 4009) {
+				}if (event.code === 4006 || event.code === 4009) {
 					err = new Error("Invalid session");
-				} else if (code === 4007) {
+				} else if (event.code === 4007) {
 					_this38.seq = 0;
 					err = new Error("Invalid sequence number");
-				} else if (code === 4008) {
+				} else if (event.code === 4008) {
 					err = new Error("Gateway connection was ratelimited");
-				} else if (code === 4010) {
+				} else if (event.code === 4010) {
 					err = new Error("Invalid shard key");
 				}
 				if (err) {
diff --git a/src/Client/InternalClient.js b/src/Client/InternalClient.js
index ff7dbe2c8..26668f5fd 100644
--- a/src/Client/InternalClient.js
+++ b/src/Client/InternalClient.js
@@ -1569,30 +1569,30 @@ export default class InternalClient {
 			self.sendWS(data);
 		};
 
-		this.websocket.onclose = (code) => {
+		this.websocket.onclose = (event) => {
 			self.websocket = null;
 			self.state = ConnectionState.DISCONNECTED;
-			if(code) {
-                client.emit("warn", "WS close: " + code);
+			if(event && event.code) {
+                client.emit("warn", "WS close: " + event.code);
                 var err;
-                if(code === 4001) {
+                if(event.code === 4001) {
                     err = new Error("Gateway received invalid OP code");
-                } else if(code === 4005) {
+                } else if(event.code === 4005) {
                     err = new Error("Gateway received invalid message");
-                } else if(code === 4003) {
+                } else if(event.code === 4003) {
                     err = new Error("Not authenticated");
-                } else if(code === 4004) {
+                } else if(event.code === 4004) {
                     err = new Error("Authentication failed");
-                } else if(code === 4005) {
+                } else if(event.code === 4005) {
                     err = new Error("Already authenticated");
-                } if(code === 4006 || code === 4009) {
+                } if(event.code === 4006 || event.code === 4009) {
                     err = new Error("Invalid session");
-                } else if(code === 4007) {
+                } else if(event.code === 4007) {
                     this.seq = 0;
                     err = new Error("Invalid sequence number");
-                } else if(code === 4008) {
+                } else if(event.code === 4008) {
                     err = new Error("Gateway connection was ratelimited");
-                } else if(code === 4010) {
+                } else if(event.code === 4010) {
                     err = new Error("Invalid shard key");
                 }
                 if(err) {