diff --git a/Library/PTPusherConnection.h b/Library/PTPusherConnection.h
index fa9768e1..6be29718 100644
--- a/Library/PTPusherConnection.h
+++ b/Library/PTPusherConnection.h
@@ -7,7 +7,6 @@
 //
 
 #import <Foundation/Foundation.h>
-#import "SRWebSocket.h"
 #import "PTPusherMacros.h"
 
 @class PTPusherConnection;
@@ -32,7 +31,7 @@ typedef enum {
   PTPusherConnectionConnected
 } PTPusherConnectionState;
 
-@interface PTPusherConnection : NSObject <SRWebSocketDelegate>
+@interface PTPusherConnection : NSObject
 
 @property (nonatomic, weak) id<PTPusherConnectionDelegate> delegate;
 
diff --git a/Library/PTPusherConnection.m b/Library/PTPusherConnection.m
index b2023ea3..24ed5dbf 100644
--- a/Library/PTPusherConnection.m
+++ b/Library/PTPusherConnection.m
@@ -16,7 +16,7 @@
 NSString *const PTPusherConnectionPingEvent        = @"pusher:ping";
 NSString *const PTPusherConnectionPongEvent        = @"pusher:pong";
 
-@interface PTPusherConnection ()
+@interface PTPusherConnection () <SRWebSocketDelegate>
 @property (nonatomic, copy) NSString *socketID;
 @property (nonatomic, assign) PTPusherConnectionState state;
 @property (nonatomic, strong) NSTimer *pingTimer;
diff --git a/Library/PTPusherMockConnection.m b/Library/PTPusherMockConnection.m
index 423ef47d..ec5dd341 100644
--- a/Library/PTPusherMockConnection.m
+++ b/Library/PTPusherMockConnection.m
@@ -9,6 +9,10 @@
 #import "PTPusherMockConnection.h"
 #import "PTJSON.h"
 #import "PTPusherEvent.h"
+#import "SRWebSocket.h"
+
+@interface PTPusherConnection () <SRWebSocketDelegate>
+@end
 
 @implementation PTPusherMockConnection {
   NSMutableArray *sentClientEvents;