use new AllFunctionUnits.get_fu function

diff --git a/src/soc/simple/test/test_core.py b/src/soc/simple/test/test_core.py
index bb8cb93dd8edcd5701170b55302687bdb6cef1fe..2f588f2600e71fc23c3f5b39e335cd259390a6fb 100644 (file)
--- a/src/soc/simple/test/test_core.py
+++ b/src/soc/simple/test/test_core.py
@@ -41,7 +41,7 @@ from openpower.util import spr_to_fast_reg
 mmu_sprs = ["PRTBL", "DSISR", "DAR", "PIDR"]
 
 def set_mmu_spr(name, i, val, core): #important keep pep8 formatting
-        fsm = core.fus.fus["mmu0"].alu
+        fsm = core.fus.get_fu("mmu0").alu
         yield fsm.mmu.l_in.mtspr.eq(1)
         yield fsm.mmu.l_in.sprn.eq(i)
         yield fsm.mmu.l_in.rs.eq(val)