diff --git a/src/wp-admin/includes/template.php b/src/wp-admin/includes/template.php
index fbd77136da..1171694bac 100644
--- a/src/wp-admin/includes/template.php
+++ b/src/wp-admin/includes/template.php
@@ -2237,7 +2237,7 @@ function get_post_states( $post ) {
  * Outputs the attachment media states as HTML.
  *
  * @since 3.2.0
- * @since 5.6.0 Added the `$echo` parameter.
+ * @since 5.6.0 Added the `$echo` parameter and a return value.
  *
  * @param WP_Post $post The attachment post to retrieve states for.
  * @param bool    $echo Optional. Whether to echo the post states as an HTML string. Default true.