diff --git a/src/mantle/framework/console/generators/stubs/factory-post.stub b/src/mantle/framework/console/generators/stubs/factory-post.stub
index a36a9ac68..62f0e851b 100644
--- a/src/mantle/framework/console/generators/stubs/factory-post.stub
+++ b/src/mantle/framework/console/generators/stubs/factory-post.stub
@@ -15,13 +15,13 @@ use App\Models\{{ class }};
 /**
  * {{ class }} Factory
  *
- * @extends \Mantle\Database\Factory\Post_Factory<\App\Models\{{ class }}>
+ * @extends \Mantle\Database\Factory\Post_Factory<\App\Models\{{ class }}, \WP_Post, {{ class }}>
  */
 class {{ class }}_Factory extends \Mantle\Database\Factory\Post_Factory {
 	/**
 	 * Model to use when creating objects.
 	 *
-	 * @var class-string<\Mantle\Database\Model\Model>
+	 * @var class-string<{{ class }}>
 	 */
 	protected string $model = {{ class }}::class;
 
diff --git a/src/mantle/framework/console/generators/stubs/factory-term.stub b/src/mantle/framework/console/generators/stubs/factory-term.stub
index 03fe50e72..9e08af1e3 100644
--- a/src/mantle/framework/console/generators/stubs/factory-term.stub
+++ b/src/mantle/framework/console/generators/stubs/factory-term.stub
@@ -15,7 +15,7 @@ use App\Models\{{ class }};
 /**
  * {{ class }} Factory
  *
- * @extends \Mantle\Database\Factory\Term_Factory<\App\Models\{{ class }}>
+ * @extends \Mantle\Database\Factory\Term_Factory<\App\Models\{{ class }}, \WP_Term, {{ class }}>
  */
 class {{ class }}_Factory extends \Mantle\Database\Factory\Term_Factory {
 	/**