replace sim._state.timeline.now with sim._engine.now

[soc.git] / src / soc / fu / div / test / test_fsm.py

diff --git a/src/soc/fu/div/test/test_fsm.py b/src/soc/fu/div/test/test_fsm.py
index fce8418bbf8272cdbecbcb9faacda350d97285ca..f18d5d7e41c269adc2da8649cc736293c1ba0dbc 100644 (file)
--- a/src/soc/fu/div/test/test_fsm.py
+++ b/src/soc/fu/div/test/test_fsm.py
@@ -235,8 +235,8 @@ class TestDivState(unittest.TestCase):
                 try:
                     # FIXME(programmerjake): replace with public API
                     # see https://github.com/nmigen/nmigen/issues/443
-                    now = sim._state.timeline.now
-                except KeyError:
+                    now = sim._engine.now
+                except AttributeError:
                     pass
                 if divisor != 0:
                     quotient = dividend // divisor