diff --git a/tests/includes/class-test-mailer.php b/tests/includes/class-test-mailer.php
index 21cfb406f..a424be080 100644
--- a/tests/includes/class-test-mailer.php
+++ b/tests/includes/class-test-mailer.php
@@ -308,6 +308,11 @@ function ( $args ) {
 		wp_delete_user( $user_id );
 	}
 
+	/**
+	 * Data provider for direct message notification text.
+	 *
+	 * @return array
+	 */
 	public function direct_message_text_provider() {
 		return array(
 			'HTML entities' => array(
@@ -324,6 +329,9 @@ public function direct_message_text_provider() {
 	/**
 	 * Test direct message notification text.
 	 *
+	 * @param string $text     Text to test.
+	 * @param string $expected Expected result.
+	 *
 	 * @covers ::direct_message
 	 * @dataProvider direct_message_text_provider
 	 */