-from nmigen import Elaboratable, Module, Signal, Record
+from nmigen import Elaboratable, Module, Signal, Record, Const, Mux
 from nmigen.utils import log2_int
 
 from ..cache import L1Cache
-from ..wishbone import wishbone_layout
+from ..wishbone import wishbone_layout, WishboneArbiter, Cycle
 
 
 __all__ = ["PCSelector", "FetchUnitInterface", "BareFetchUnit", "CachedFetchUnit"]
 
-from nmigen import Elaboratable, Module, Signal, Record, Cat
+from nmigen import Elaboratable, Module, Signal, Record, Cat, Const, Mux
 from nmigen.utils import log2_int
 from nmigen.lib.fifo import SyncFIFO
 
 from ..cache import L1Cache
 from ..isa import Funct3
-from ..wishbone import wishbone_layout
+from ..wishbone import wishbone_layout, WishboneArbiter, Cycle
 
 
 __all__ = ["DataSelector", "LoadStoreUnitInterface", "BareLoadStoreUnit", "CachedLoadStoreUnit"]