diff --git a/src/Console/stubs/partial.stub b/src/Console/stubs/partial.stub
index 36463d3f..3184b0e6 100644
--- a/src/Console/stubs/partial.stub
+++ b/src/Console/stubs/partial.stub
@@ -10,7 +10,7 @@ class DummyClass extends Partial
     /**
      * The partial field group.
      *
-     * @return array
+     * @return \StoutLogic\AcfBuilder\FieldsBuilder
      */
     public function fields()
     {