diff --git a/src/Http1TunnelConnector.php b/src/Http1TunnelConnector.php
index 4cade76..fc3f348 100644
--- a/src/Http1TunnelConnector.php
+++ b/src/Http1TunnelConnector.php
@@ -5,6 +5,7 @@
 use Amp\Cancellation;
 use Amp\ForbidCloning;
 use Amp\ForbidSerialization;
+use Amp\Http\HttpMessage;
 use Amp\Http\Tunnel\Internal\TunnelSocket;
 use Amp\NullCancellation;
 use Amp\Socket\ConnectContext;
@@ -14,11 +15,17 @@
 use function Amp\now;
 use function Amp\Socket\socketConnector;
 
+/**
+ * @psalm-import-type HeaderParamArrayType from HttpMessage
+ */
 final class Http1TunnelConnector implements SocketConnector
 {
     use ForbidCloning;
     use ForbidSerialization;
 
+    /**
+     * @param HeaderParamArrayType $customHeaders
+     */
     public function __construct(
         private readonly string $proxyAddress,
         private readonly array $customHeaders = [],
diff --git a/src/Https1TunnelConnector.php b/src/Https1TunnelConnector.php
index 78f3ffb..42492b7 100644
--- a/src/Https1TunnelConnector.php
+++ b/src/Https1TunnelConnector.php
@@ -5,6 +5,7 @@
 use Amp\Cancellation;
 use Amp\ForbidCloning;
 use Amp\ForbidSerialization;
+use Amp\Http\HttpMessage;
 use Amp\Http\Tunnel\Internal\TunnelSocket;
 use Amp\NullCancellation;
 use Amp\Socket\ClientTlsContext;
@@ -20,11 +21,17 @@
 use function Amp\Socket\listen;
 use function Amp\Socket\socketConnector;
 
+/**
+ * @psalm-import-type HeaderParamArrayType from HttpMessage
+ */
 final class Https1TunnelConnector implements SocketConnector
 {
     use ForbidCloning;
     use ForbidSerialization;
 
+    /**
+     * @param HeaderParamArrayType $customHeaders
+     */
     public function __construct(
         private readonly string $proxyAddress,
         private readonly ClientTlsContext $proxyTlsContext,