diff --git a/lib/PostHog.php b/lib/PostHog.php
index b872dcc..65ab781 100644
--- a/lib/PostHog.php
+++ b/lib/PostHog.php
@@ -178,7 +178,7 @@ public static function getFeatureFlag(
      * @param array $groups
      * @param array $personProperties
      * @param array $groupProperties
-     * @return boolean | string
+     * @return array
      * @throws Exception
      */
     public static function getAllFlags(