diff --git a/test/test_packet2.py b/test/test_packet2.py
index bd315d4d7..082a97083 100644
--- a/test/test_packet2.py
+++ b/test/test_packet2.py
@@ -12,7 +12,7 @@ from migen import *
 from litex.soc.interconnect.stream import *
 from litex.soc.interconnect.packet import *
 
-from .test_stream import StreamPacket, stream_inserter, stream_collector, compare_packets
+from .test_stream2 import StreamPacket, stream_inserter, stream_collector, compare_packets
 
 def mask_last_be(dw, data, last_be):
     masked_data = 0