diff --git a/tests/TestCase.php b/tests/TestCase.php
index 003278e..cfd64e4 100644
--- a/tests/TestCase.php
+++ b/tests/TestCase.php
@@ -2,12 +2,8 @@
 
 namespace BeyondCode\LaravelWebSockets\Tests;
 
-use Orchestra\Testbench\BrowserKit\TestCase as BaseTestCase;
-
-abstract class TestCase extends BaseTestCase
+abstract class TestCase extends WebSocketsTestCase
 {
-    use Concerns\TestsWebSockets;
-
     /**
      * {@inheritdoc}
      */
diff --git a/tests/Concerns/TestsWebSockets.php b/tests/WebSocketsTestCase.php
similarity index 89%
rename from tests/Concerns/TestsWebSockets.php
rename to tests/WebSocketsTestCase.php
index 86bc9ab..e90ece0 100644
--- a/tests/Concerns/TestsWebSockets.php
+++ b/tests/WebSocketsTestCase.php
@@ -1,19 +1,10 @@
 <?php
 
-namespace BeyondCode\LaravelWebSockets\Tests\Concerns;
+namespace BeyondCode\LaravelWebSockets\Tests;
 
-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;
+use Orchestra\Testbench\BrowserKit\TestCase as BaseTestCase;
 
-trait TestsWebSockets
+class WebSocketsTestCase extends BaseTestCase
 {
     /**
      * A test Pusher server.