diff --git a/src/danog/MadelineProto/Stream/Common/BufferedRawStream.php b/src/danog/MadelineProto/Stream/Common/BufferedRawStream.php
index 129380e9..56defd56 100644
--- a/src/danog/MadelineProto/Stream/Common/BufferedRawStream.php
+++ b/src/danog/MadelineProto/Stream/Common/BufferedRawStream.php
@@ -24,6 +24,7 @@ use danog\MadelineProto\Exception;
 use danog\MadelineProto\Stream\Async\RawStream;
 use danog\MadelineProto\Stream\ConnectionContext;
 use function Amp\Socket\connect;
+use Amp\ByteStream\ClosedException;
 
 /**
  * Buffered raw stream.