diff --git a/rector.php b/rector.php
index 583f7c5..6495e54 100644
--- a/rector.php
+++ b/rector.php
@@ -113,6 +113,7 @@
         ],
         JsonThrowOnErrorRector::class => [
             __DIR__.'/app/Generators/OpenAIGenerator.php',
+            __DIR__.'/app/Generators/ErnieBotGenerator.php',
         ],
         StaticClosureRector::class => [
             __DIR__.'/tests',
diff --git a/tests/Pest.php b/tests/Pest.php
index 2c59328..8f7285d 100644
--- a/tests/Pest.php
+++ b/tests/Pest.php
@@ -91,7 +91,7 @@ function setup_http_fake(): void
                         break;
                     case 'invalid':
                         $text = <<<'json'
-                            [{"id":1,"subject":"Fix(OpenAIGenerator): Debugging output","body":"- Add var_dump() for debugging output- Add var_dump() for stream response"},{"id":2,"subject":"Refactor(OpenAIGenerator): Error handling","body":"- Check for error response in JSON- Handle error response"},{"id":3,"subject":"Docs(OpenAIGenerator): Update documentation","body":"- Update documentation for OpenAIGenerator class"}]
+                            [{id":1,"subject":"Fix(OpenAIGenerator): Debugging output","body":"- Add var_dump() for debugging output- Add var_dump() for stream response"},{"id":2,"subject":"Refactor(OpenAIGenerator): Error handling","body":"- Check for error response in JSON- Handle error response"},{"id":3,"subject":"Docs(OpenAIGenerator): Update documentation","body":"- Update documentation for OpenAIGenerator class"}]
                             json;
 
                         break;