diff --git a/code/data_gen.cpp b/code/data_gen.cpp
index 0d00122..a39b5a0 100644
--- a/code/data_gen.cpp
+++ b/code/data_gen.cpp
@@ -386,8 +386,8 @@ namespace node_relations{
                 if (i != j) {
                     int new_x = order_list[i], new_y = order_list[j];
                     for (int k = 0; k < 7; k++) {
-                        if ((graph_2.pair_relations[new_x][new_y]>>k)&1)
-                            graph_1.pair_relation_count[i][j][k]++;
+                        if ((graph_2.pair_relations[i][j]>>k)&1)
+                            graph_1.pair_relation_count[new_x][new_y][k]++;
                     }
                 }
     }