diff --git a/src/CountryLoader.php b/src/CountryLoader.php
index f5b56c7..a8860d9 100644
--- a/src/CountryLoader.php
+++ b/src/CountryLoader.php
@@ -23,7 +23,7 @@ class CountryLoader
      *
      * @throws CountryLoaderException
      *
-     * @return Country|array
+     * @return ($hydrate is true ? Country : array)
      */
     public static function country($code, $hydrate = true)
     {
@@ -44,7 +44,7 @@ public static function country($code, $hydrate = true)
      *
      * @throws CountryLoaderException
      *
-     * @return array
+     * @return ($hydrate is true ? array<string, Country> : array<string, array>)
      */
     public static function countries($longlist = false, $hydrate = false)
     {
diff --git a/src/helpers.php b/src/helpers.php
index a536bfa..ad6926f 100644
--- a/src/helpers.php
+++ b/src/helpers.php
@@ -13,7 +13,7 @@
      * @param string $code
      * @param bool   $hydrate
      *
-     * @return Country|array
+     * @return ($hydrate is true ? Country : array)
      */
     function country($code, $hydrate = true)
     {
@@ -28,7 +28,7 @@ function country($code, $hydrate = true)
      * @param bool $longlist
      * @param bool $hydrate
      *
-     * @return array
+     * @return ($hydrate is true ? array<string, Country> : array<string, array>)
      */
     function countries($longlist = false, $hydrate = false)
     {