diff --git a/README.md b/README.md
index ea3cbb0..6cb5722 100644
--- a/README.md
+++ b/README.md
@@ -98,15 +98,15 @@ while True:
         for row in row_change.rowDatas:
             format_data = dict()
             if event_type == EntryProtocol_pb2.EventType.DELETE:
-                for column in row.beforeColumns:
-                    format_data = {
-                        column.name: column.value
-                    }
+                format_data = {
+                    column.name: column.value
+                    for column in row.beforeColumns
+                }
             elif event_type == EntryProtocol_pb2.EventType.INSERT:
-                for column in row.afterColumns:
-                    format_data = {
-                        column.name: column.value
-                    }
+                format_data = {
+                    column.name: column.value
+                    for column in row.afterColumns
+                }
             else:
                 format_data['before'] = format_data['after'] = dict()
                 for column in row.beforeColumns:
diff --git a/canal/example.py b/canal/example.py
index de68f90..771d3e4 100644
--- a/canal/example.py
+++ b/canal/example.py
@@ -28,15 +28,15 @@
         for row in row_change.rowDatas:
             format_data = dict()
             if event_type == EntryProtocol_pb2.EventType.DELETE:
-                for column in row.beforeColumns:
-                    format_data = {
-                        column.name: column.value
-                    }
+                format_data = {
+                    column.name: column.value
+                    for column in row.beforeColumns
+                }
             elif event_type == EntryProtocol_pb2.EventType.INSERT:
-                for column in row.afterColumns:
-                    format_data = {
-                        column.name: column.value
-                    }
+                format_data = {
+                    column.name: column.value
+                    for column in row.afterColumns
+                }
             else:
                 format_data['before'] = format_data['after'] = dict()
                 for column in row.beforeColumns:
@@ -52,4 +52,4 @@
             print(data)
     time.sleep(1)
 
-client.disconnect()
\ No newline at end of file
+client.disconnect()