diff --git a/examples/sharding/driver.py b/examples/sharding/driver.py
index 4989d49..7dd4669 100644
--- a/examples/sharding/driver.py
+++ b/examples/sharding/driver.py
@@ -72,7 +72,9 @@ async def main():
     
     # Define topic map
     topic_map = {
-        "1": [0, 1, 2, 3, 4, 5]
+        "1": [0, 1],
+        "2": [2, 3],
+        "3": [4, 5]
     }
 
     topics = topic_map.keys()