diff --git a/tests/WebSocketsTestCase.php b/tests/Concerns/TestsWebSockets.php
similarity index 89%
rename from tests/WebSocketsTestCase.php
rename to tests/Concerns/TestsWebSockets.php
index e90ece0..86bc9ab 100644
--- a/tests/WebSocketsTestCase.php
+++ b/tests/Concerns/TestsWebSockets.php
@@ -1,10 +1,19 @@
 <?php
 
-namespace BeyondCode\LaravelWebSockets\Tests;
+namespace BeyondCode\LaravelWebSockets\Tests\Concerns;
 
-use Orchestra\Testbench\BrowserKit\TestCase as BaseTestCase;
+use BeyondCode\LaravelWebSockets\Facades\StatisticsLogger;
+use BeyondCode\LaravelWebSockets\PubSub\ReplicationInterface;
+use BeyondCode\LaravelWebSockets\Statistics\Drivers\StatisticsDriver;
+use BeyondCode\LaravelWebSockets\Tests\Mocks\Connection;
+use BeyondCode\LaravelWebSockets\Tests\Mocks\FakeMemoryStatisticsLogger;
+use BeyondCode\LaravelWebSockets\Tests\Mocks\Message;
+use BeyondCode\LaravelWebSockets\WebSockets\Channels\ChannelManager;
+use GuzzleHttp\Psr7\Request;
+use Ratchet\ConnectionInterface;
+use React\EventLoop\Factory as LoopFactory;
 
-class WebSocketsTestCase extends BaseTestCase
+trait TestsWebSockets
 {
     /**
      * A test Pusher server.
diff --git a/tests/TestCase.php b/tests/TestCase.php
index cfd64e4..003278e 100644
--- a/tests/TestCase.php
+++ b/tests/TestCase.php
@@ -2,8 +2,12 @@
 
 namespace BeyondCode\LaravelWebSockets\Tests;
 
-abstract class TestCase extends WebSocketsTestCase
+use Orchestra\Testbench\BrowserKit\TestCase as BaseTestCase;
+
+abstract class TestCase extends BaseTestCase
 {
+    use Concerns\TestsWebSockets;
+
     /**
      * {@inheritdoc}
      */