diff --git a/tests/Channels/PresenceChannelReplicationTest.php b/tests/Channels/PresenceChannelReplicationTest.php
index 972a8bf..0d605f7 100644
--- a/tests/Channels/PresenceChannelReplicationTest.php
+++ b/tests/Channels/PresenceChannelReplicationTest.php
@@ -2,6 +2,7 @@
 
 namespace BeyondCode\LaravelWebSockets\Tests\Channels;
 
+use BeyondCode\LaravelWebSockets\Tests\Mocks\Message;
 use BeyondCode\LaravelWebSockets\Tests\TestCase;
 
 class PresenceChannelReplicationTest extends TestCase