1 files changed, 2 insertions, 2 deletions
diff --git a/tcp-reassembly.py b/tcp-reassembly.py
index 2ae570d..21abf9f 100755
--- a/tcp-reassembly.py
+++ b/tcp-reassembly.py
@@ -80,8 +80,8 @@ for i, (pktno, pkt) in enumerate(mypackets):
         pkt.load += pkt[Padding].load
         del pkt[Padding]
         pkt[IP].len = None
-        mypackets[i] = Ether(bytes(pkt))
-        assert mypackets[i][IP].len != 0
+        mypackets[i] = pktno, Ether(bytes(pkt))
+        assert mypackets[i][1][IP].len != 0
 
 
 class SegmentInfo(object):