diff --git a/examples/kafka_consumer.cpp b/examples/kafka_consumer.cpp
index 46c7f3d..6d82289 100644
--- a/examples/kafka_consumer.cpp
+++ b/examples/kafka_consumer.cpp
@@ -13,6 +13,7 @@ using std::endl;
 using cppkafka::Consumer;
 using cppkafka::Configuration;
 using cppkafka::Message;
+using cppkafka::TopicPartitionList;
 
 namespace po = boost::program_options;
 
@@ -60,6 +61,16 @@ int main(int argc, char* argv[]) {
     // Create the consumer
     Consumer consumer(config);
 
+    // Print the assigned partitions on assignment
+    consumer.set_assignment_callback([](const TopicPartitionList& partitions) {
+        cout << "Got assigned: " << partitions << endl;
+    });
+
+    // Print the revoked partitions on revocation
+    consumer.set_revocation_callback([](const TopicPartitionList& partitions) {
+        cout << "Got revoked: " << partitions << endl;
+    });
+
     // Subscribe to the topic
     consumer.subscribe({ topic_name });