diff --git a/codegen/migrate.py b/codegen/migrate.py
index 75cbe122..304351d3 100644
--- a/codegen/migrate.py
+++ b/codegen/migrate.py
@@ -47,10 +47,11 @@ print()
 changed_packets: dict[PacketIdentifier, int] = {}
 for old_packet, old_packet_name in old_packets.items():
     for new_packet, new_packet_name in new_packets.items():
-        if old_packet == new_packet and old_packet.packet_id != new_packet.packet_id:
+        if old_packet.direction == new_packet.direction and old_packet.state == new_packet.state and old_packet.packet_id != new_packet.packet_id:
             changed_packets[old_packet] = new_packet.packet_id
             print('Changed packet id:', old_packet, '->',
                   new_packet, f'({new_packet_name})')
+            break
 for (direction, state), packets in group_packets(list(changed_packets.keys())).items():
     lib.code.packet.remove_packet_ids(packets, direction, state)