diff --git a/artiq/gateware/rtio/sed/lane_distributor.py b/artiq/gateware/rtio/sed/lane_distributor.py
index ddd19ee33..9b3a17e64 100644
--- a/artiq/gateware/rtio/sed/lane_distributor.py
+++ b/artiq/gateware/rtio/sed/lane_distributor.py
@@ -3,6 +3,9 @@ from migen import *
 from artiq.gateware.rtio import cri
 
 
+__all__ = ["LaneDistributor"]
+
+
 def layout_lane_io(seqn_width, layout_payload):
     return [
         ("we", 1, DIR_M_TO_S),
diff --git a/artiq/gateware/rtio/sed/output_driver.py b/artiq/gateware/rtio/sed/output_driver.py
index ad9a6c2e9..5ee24753e 100644
--- a/artiq/gateware/rtio/sed/output_driver.py
+++ b/artiq/gateware/rtio/sed/output_driver.py
@@ -7,6 +7,9 @@ from artiq.gateware.rtio import rtlink
 from artiq.gateware.rtio.sed.output_network import OutputNetwork
 
 
+__all__ = ["OutputDriver"]
+
+
 class OutputDriver(Module):
     def __init__(self, channels, lane_count, seqn_width):
         self.collision = Signal()