diff --git a/tests/src/Traits/ApiTestTrait.php b/tests/src/Traits/ApiTestTrait.php
index 4585de3..a9c14d6 100644
--- a/tests/src/Traits/ApiTestTrait.php
+++ b/tests/src/Traits/ApiTestTrait.php
@@ -45,7 +45,7 @@ protected function createMockHttpClient(array $responses) : Client {
   /**
    * Overrides the default 'http_client_factory' service with mock.
    *
-   * @param \Psr\Http\Message\ResponseInterface[] $responses
+   * @param \Psr\Http\Message\ResponseInterface[]|\GuzzleHttp\Exception\GuzzleException[] $responses
    *   The expected responses.
    *
    * @return \GuzzleHttp\Client