Import Delay in gram tests

diff --git a/gram/test/utils.py b/gram/test/utils.py
index 67124296d3ca276ab4022a7782a2702dfe14b285..aa22a39678d586db01898a4fcc507816bf45e335 100644 (file)
--- a/gram/test/utils.py
+++ b/gram/test/utils.py
@@ -9,13 +9,13 @@ import warnings
 from contextlib import contextmanager
 
 from nmigen import *
-from nmigen.back.pysim import *
+from nmigen.sim.pysim import *
 from nmigen.hdl.ir import Fragment
 from nmigen.back import rtlil
 from nmigen._toolchain import require_tool
 
 
-__all__ = ["FHDLTestCase", "runSimulation", "wb_read", "wb_write", "PulseCounter"]
+__all__ = ["FHDLTestCase", "runSimulation", "wb_read", "wb_write", "PulseCounter", "Delay"]
 
 def runSimulation(module, process, vcd_filename="anonymous.vcd", clock=1e-6):
     sim = Simulator(module)