diff --git a/src/Facades/GrokAI.php b/src/Facades/GrokAI.php
index edcbd8a..6a0f05e 100644
--- a/src/Facades/GrokAI.php
+++ b/src/Facades/GrokAI.php
@@ -3,10 +3,11 @@
 namespace GrokPHP\Laravel\Facades;
 
 use GrokPHP\Client\Clients\Vision;
+use GrokPHP\Laravel\Support\GrokResponse;
 use Illuminate\Support\Facades\Facade;
 
 /**
- * @method static array chat(array $messages, ?\GrokPHP\Client\Config\ChatOptions $options = null)
+ * @method static GrokResponse chat(array $messages, ?\GrokPHP\Client\Config\ChatOptions $options = null)
  * @method static Vision vision()
  */
 class GrokAI extends Facade