diff --git a/psalm-baseline.xml b/psalm-baseline.xml
index 9c7b310621ed..f74923641a07 100644
--- a/psalm-baseline.xml
+++ b/psalm-baseline.xml
@@ -787,16 +787,6 @@
       <code>$locale</code>
     </PossiblyNullArgument>
   </file>
-  <file src="src/Network/Socket.php">
-    <InvalidArgument occurrences="1">
-      <code>[$this, '_connectionErrorHandler']</code>
-    </InvalidArgument>
-    <PossiblyNullArgument occurrences="3">
-      <code>$this-&gt;connection</code>
-      <code>$this-&gt;connection</code>
-      <code>$this-&gt;connection</code>
-    </PossiblyNullArgument>
-  </file>
   <file src="src/ORM/BehaviorRegistry.php">
     <MoreSpecificImplementedParamType occurrences="1">
       <code>$class</code>
diff --git a/src/Network/Socket.php b/src/Network/Socket.php
index d40d6f294b9b..6d9d400058eb 100644
--- a/src/Network/Socket.php
+++ b/src/Network/Socket.php
@@ -154,6 +154,7 @@ public function connect(): bool
             $connectAs |= STREAM_CLIENT_PERSISTENT;
         }
 
+        /** @psalm-suppress InvalidArgument */
         set_error_handler([$this, '_connectionErrorHandler']);
         $remoteSocketTarget = $scheme . $this->_config['host'];
         $port = (int)$this->_config['port'];
@@ -185,6 +186,7 @@ public function connect(): bool
 
         $this->connected = is_resource($this->connection);
         if ($this->connected) {
+            /** @psalm-suppress PossiblyNullArgument */
             stream_set_timeout($this->connection, $this->_config['timeout']);
         }
 
@@ -364,6 +366,7 @@ public function write(string $data): int
         $totalBytes = strlen($data);
         $written = 0;
         while ($written < $totalBytes) {
+            /** @psalm-suppress PossiblyNullArgument */
             $rv = fwrite($this->connection, substr($data, $written));
             if ($rv === false || $rv === 0) {
                 return $written;
@@ -387,6 +390,7 @@ public function read(int $length = 1024): ?string
             return null;
         }
 
+        /** @psalm-suppress PossiblyNullArgument */
         if (!feof($this->connection)) {
             $buffer = fread($this->connection, $length);
             $info = stream_get_meta_data($this->connection);