From f6e4259019b242ab462a51eaeba3201335d92326 Mon Sep 17 00:00:00 2001
From: kieumaverick92 <52841526+kieumaverick92@users.noreply.github.com>
Date: Mon, 26 Aug 2019 16:01:45 -0400
Subject: [PATCH] Update Scaledrone.swift

---
 scaledrone-swift/Scaledrone.swift | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/scaledrone-swift/Scaledrone.swift b/scaledrone-swift/Scaledrone.swift
index 98f09f3..0959ef3 100644
--- a/scaledrone-swift/Scaledrone.swift
+++ b/scaledrone-swift/Scaledrone.swift
@@ -79,7 +79,7 @@ public class Scaledrone: WebSocketDelegate {
     
     // MARK: Websocket Delegate Methods.
     
-    public func websocketDidConnect(socket: WebSocketClient) {
+    public func websocketDidConnect(socket: WebSocket) {
         var msg = [
             "type": "handshake",
             "channel": self.channelID,
@@ -94,11 +94,11 @@ public class Scaledrone: WebSocketDelegate {
         self.send(msg)
     }
     
-    public func websocketDidDisconnect(socket: WebSocketClient, error: Error?) {
+    public func websocketDidDisconnect(socket: WebSocket, error: NSError?) {
         delegate?.scaledroneDidDisconnect(scaledrone: self, error: error)
     }
     
-    public func websocketDidReceiveMessage(socket: WebSocketClient, text: String) {
+    public func websocketDidReceiveMessage(socket: WebSocket, text: String) {
         var dic = convertJSONMessageToDictionary(text: text)
         
         if let error = dic["error"] as? String {
@@ -177,7 +177,7 @@ public class Scaledrone: WebSocketDelegate {
         room.delegate?.scaledroneRoomDidReceiveMessage(room: room, message: messageDic["message"] as Any, member: member)
     }
     
-    public func websocketDidReceiveData(socket: WebSocketClient, data: Data) {
+    public func websocketDidReceiveData(socket: WebSocket, data: Data) {
         print("Should not have received any data: \(data.count)")
     }