from nmigen import Elaboratable, Signal, Module
from nmigen.back.pysim import Simulator, Fragment
-from .. import csr
-from ..csr.wishbone import *
+from nmigen_soc import csr
+from nmigen_soc.csr.wishbone import WishboneCSRBridge
class MockRegister(Elaboratable):