diff --git a/examples/client-communication/server.php b/examples/client-communication/server.php
index 7d586ba..42e5058 100644
--- a/examples/client-communication/server.php
+++ b/examples/client-communication/server.php
@@ -16,7 +16,7 @@
 use Mcp\Schema\Enum\LoggingLevel;
 use Mcp\Schema\ServerCapabilities;
 use Mcp\Server;
-use Mcp\Server\ClientGateway;
+use Mcp\Server\RequestContext;
 use Mcp\Server\Session\FileSessionStore;
 
 $server = Server::builder()
@@ -27,7 +27,8 @@
     ->setCapabilities(new ServerCapabilities(logging: true, tools: true))
     ->setDiscovery(__DIR__)
     ->addTool(
-        function (string $dataset, ClientGateway $client): array {
+        function (RequestContext $context, string $dataset): array {
+            $client = $context->getClientGateway();
             $client->log(LoggingLevel::Info, sprintf('Running quality checks on dataset "%s"', $dataset));
 
             $tasks = [