2 files changed, 2 insertions, 5 deletions
diff --git a/test/stubs/test_connection.rb b/test/stubs/test_connection.rb
index 1c353f9ee3..384abc5e76 100644
--- a/test/stubs/test_connection.rb
+++ b/test/stubs/test_connection.rb
@@ -18,9 +18,4 @@ class TestConnection
   def last_transmission
     @transmissions.last
   end
-
-  # Disable async in tests
-  def send_async(method, *arguments)
-    send method, *arguments
-  end
 end
diff --git a/test/stubs/test_server.rb b/test/stubs/test_server.rb
index aec80859c3..2a7ac3e927 100644
--- a/test/stubs/test_server.rb
+++ b/test/stubs/test_server.rb
@@ -1,6 +1,8 @@
 require 'ostruct'
 
 class TestServer
+  include ActionCable::Server::Connections
+
   attr_reader :logger, :config
 
   def initialize