diff --git a/docs/src/main/asciidoc/kafka/kafka_partitions.adoc b/docs/src/main/asciidoc/kafka/kafka_partitions.adoc
index 728d7d656..6b53250ff 100644
--- a/docs/src/main/asciidoc/kafka/kafka_partitions.adoc
+++ b/docs/src/main/asciidoc/kafka/kafka_partitions.adoc
@@ -74,15 +74,15 @@ public class KafkaPartitionConsumerApplication {
 
     public static void main(String[] args) {
         new SpringApplicationBuilder(KafkaPartitionConsumerApplication.class)
-            .web(false)
+            .web(WebApplicationType.NONE)
             .run(args);
     }
 
     @Bean
     public Consumer<Message<String>> listen() {
         return message -> {
-            int partition =- message.getHeaders().get(KafkaHeaders.RECEIVED_PARTITION_ID);
-            System.out.println(in + " received from partition " + partition);
+            int partition = (int) message.getHeaders().get(KafkaHeaders.RECEIVED_PARTITION);
+            System.out.println(message + " received from partition " + partition);
         };
     }