diff --git a/tests/ClientTest.php b/tests/ClientTest.php
index c1a361a..3c68942 100644
--- a/tests/ClientTest.php
+++ b/tests/ClientTest.php
@@ -518,12 +518,12 @@ public function testUnconnectedClient(): void
     public function testDeprecated(): void
     {
         $client = new Client('ws://localhost:8000/my/mock/path');
-        (new ErrorHandler())->with(function () use ($client) {
+        (new ErrorHandler())->withAll(function () use ($client) {
             $this->assertNull($client->getPier());
-        }, function (ErrorException $e) {
+        }, function ($exceptions, $result) {
             $this->assertEquals(
                 'getPier() is deprecated and will be removed in future version. Use getRemoteName() instead.',
-                $e->getMessage()
+                $exceptions[0]->getMessage()
             );
         }, E_USER_DEPRECATED);
     }
diff --git a/tests/ServerTest.php b/tests/ServerTest.php
index 17e9bab..033895f 100644
--- a/tests/ServerTest.php
+++ b/tests/ServerTest.php
@@ -499,12 +499,12 @@ public function testDeprecated(): void
         MockSocket::initialize('server.construct', $this);
         $server = new Server();
         $this->assertTrue(MockSocket::isEmpty());
-        (new ErrorHandler())->with(function () use ($server) {
+        (new ErrorHandler())->withAll(function () use ($server) {
             $this->assertNull($server->getPier());
-        }, function (ErrorException $e) {
+        }, function ($exceptions, $result) {
             $this->assertEquals(
                 'getPier() is deprecated and will be removed in future version. Use getRemoteName() instead.',
-                $e->getMessage()
+                $exceptions[0]->getMessage()
             );
         }, E_USER_DEPRECATED);
     }