diff --git a/includes/api/legacy/v2/class-wc-api-products.php b/includes/api/legacy/v2/class-wc-api-products.php
index 550ab49a8c3..65edab35612 100644
--- a/includes/api/legacy/v2/class-wc-api-products.php
+++ b/includes/api/legacy/v2/class-wc-api-products.php
@@ -1834,10 +1834,10 @@ class WC_API_Products extends WC_API_Resource {
 	}
 
 	/**
-	 * Get product image as attachment
+	 * Sets product image as attachment and returns the attachment ID.
 	 *
 	 * @since 2.2
-	 * @param integer $upload
+	 * @param array $upload
 	 * @param int $id
 	 * @return int
 	 */
diff --git a/includes/api/legacy/v3/class-wc-api-products.php b/includes/api/legacy/v3/class-wc-api-products.php
index c1caa19d7a3..6e1cedea3b0 100644
--- a/includes/api/legacy/v3/class-wc-api-products.php
+++ b/includes/api/legacy/v3/class-wc-api-products.php
@@ -2395,7 +2395,7 @@ class WC_API_Products extends WC_API_Resource {
 	 * Sets product image as attachment and returns the attachment ID.
 	 *
 	 * @since 2.2
-	 * @param integer $upload
+	 * @param array $upload
 	 * @param int $id
 	 * @return int
 	 */
diff --git a/includes/cli/class-wc-cli-product.php b/includes/cli/class-wc-cli-product.php
index 1c2e5ed2c94..a41f3fd92f7 100644
--- a/includes/cli/class-wc-cli-product.php
+++ b/includes/cli/class-wc-cli-product.php
@@ -2057,10 +2057,10 @@ class WC_CLI_Product extends WC_CLI_Command {
 	}
 
 	/**
-	 * Get product image as attachment
+	 * Sets product image as attachment and returns the attachment ID.
 	 *
 	 * @since  2.5.0
-	 * @param  int $upload
+	 * @param  array $upload
 	 * @param  int $id
 	 * @return int
 	 */