diff --git a/sdks/java/io/amazon-web-services2/src/main/java/org/apache/beam/sdk/io/aws2/kinesis/KinesisIO.java b/sdks/java/io/amazon-web-services2/src/main/java/org/apache/beam/sdk/io/aws2/kinesis/KinesisIO.java
index 835bde170d33..2de4a47ebaec 100644
--- a/sdks/java/io/amazon-web-services2/src/main/java/org/apache/beam/sdk/io/aws2/kinesis/KinesisIO.java
+++ b/sdks/java/io/amazon-web-services2/src/main/java/org/apache/beam/sdk/io/aws2/kinesis/KinesisIO.java
@@ -237,9 +237,9 @@
  *
  * <pre>{@code PCollection<KV<String, byte[]>> data = ...;
  *
- * data.apply(KinesisIO.write()
+ * data.apply(KinesisIO.<KV<String, byte[]>>write()
  *     .withStreamName("streamName")
- *     .withPartitionKey(KV::getKey)
+ *     .withPartitioner(KV::getKey)
  *     .withSerializer(KV::getValue);
  * }</pre>
  *