diff --git a/connect/api/src/main/java/org/apache/kafka/connect/sink/SinkTask.java b/connect/api/src/main/java/org/apache/kafka/connect/sink/SinkTask.java
index 655c89ac670fe..71d19278287e0 100644
--- a/connect/api/src/main/java/org/apache/kafka/connect/sink/SinkTask.java
+++ b/connect/api/src/main/java/org/apache/kafka/connect/sink/SinkTask.java
@@ -130,6 +130,7 @@ public void flush(Map<TopicPartition, OffsetAndMetadata> currentOffsets) {
      *                       {@link Transformation transformations} have been applied. These can be tracked by the task
      *                       through the {@link SinkRecord#originalTopic()}, {@link SinkRecord#originalKafkaPartition()}
      *                       and {@link SinkRecord#originalKafkaOffset()} methods.
+     *                       During rebalancing, this map may contain revoked partitions.
      *
      * @return an empty map if Connect-managed offset commit is not desired, otherwise a map of offsets by topic-partition that are
      *         safe to commit. Note that the returned topic-partition to offsets map should use the original Kafka