1 files changed, 6 insertions, 2 deletions
diff --git a/lib/action_cable/server.rb b/lib/action_cable/server.rb
index d1b7e14b53..b6ea201b52 100644
--- a/lib/action_cable/server.rb
+++ b/lib/action_cable/server.rb
@@ -32,11 +32,11 @@ module ActionCable
 
         @websocket.on(:message) do |event|
           message = event.data
-          self.class.worker_pool.async.received_data(self, message) if message.is_a?(String)
+          worker_pool.async.received_data(self, message) if message.is_a?(String)
         end
 
         @websocket.on(:close) do |event|
-          self.class.worker_pool.async.cleanup_subscriptions(self)
+          worker_pool.async.cleanup_subscriptions(self)
         end
 
         @websocket.rack_response
@@ -68,6 +68,10 @@ module ActionCable
       @websocket.send data
     end
 
+    def worker_pool
+      self.class.worker_pool
+    end
+
     private
       def subscribe_channel(data)
         id_key = data['identifier']