diff --git a/includes/ThreadItemStore.php b/includes/ThreadItemStore.php
index e3555df24..d7136f205 100644
--- a/includes/ThreadItemStore.php
+++ b/includes/ThreadItemStore.php
@@ -516,13 +516,13 @@ class ThreadItemStore {
 	 * Store the thread item set.
 	 *
 	 * @param RevisionRecord $rev
-	 * @param ThreadItemSet $threadItemSet
+	 * @param ContentThreadItemSet $threadItemSet
 	 * @throws TimestampException
 	 * @throws DBError
 	 * @throws Exception
 	 * @return bool
 	 */
-	public function insertThreadItems( RevisionRecord $rev, ThreadItemSet $threadItemSet ): bool {
+	public function insertThreadItems( RevisionRecord $rev, ContentThreadItemSet $threadItemSet ): bool {
 		if ( $this->readOnlyMode->isReadOnly() ) {
 			return false;
 		}