diff --git a/src/Search.php b/src/Search.php
index da8372b0..407848d1 100644
--- a/src/Search.php
+++ b/src/Search.php
@@ -63,7 +63,7 @@ class Search
      * operations return the contents of the _source field unless you have used the
      * stored_fields parameter or if the _source field is disabled.
      *
-     * @var bool
+     * @var array|bool|string
      */
     private $source;
 
@@ -518,7 +518,7 @@ public function setSize($size)
     }
 
     /**
-     * @return bool
+     * @return array|bool|string
      */
     public function isSource()
     {
@@ -526,7 +526,7 @@ public function isSource()
     }
 
     /**
-     * @param bool $source
+     * @param array|bool|string $source
      *
      * @return $this
      */