diff --git a/README.md b/README.md
index dddb3d54e..03c30caf9 100644
--- a/README.md
+++ b/README.md
@@ -30,7 +30,7 @@ p.flush()
 **High-level Consumer:**
 
 ```python
-from confluent_kafka import Consumer
+from confluent_kafka import Consumer, KafkaError
 
 c = Consumer({'bootstrap.servers': 'mybroker', 'group.id': 'mygroup',
               'default.topic.config': {'auto.offset.reset': 'smallest'}})
@@ -40,7 +40,8 @@ while running:
     msg = c.poll()
     if not msg.error():
         print('Received message: %s' % msg.value().decode('utf-8'))
-    else:
+    elif msg.error().code() != KafkaError._PARTITION_EOF:
+        print(msg.error())
         running = False
 c.close()
 ```