fsm: NextState and NextValue should derive from _Statement

diff --git a/migen/genlib/fsm.py b/migen/genlib/fsm.py
index 5bfa3575667359fb91d7fd2413523da110d7aeef..cab83bb851c2118212551f0156464211e1d06f10 100644 (file)
--- a/migen/genlib/fsm.py
+++ b/migen/genlib/fsm.py
@@ -1,7 +1,7 @@
 from collections import OrderedDict
 
 from migen.fhdl.structure import *
-from migen.fhdl.structure import _Slice, _ArrayProxy
+from migen.fhdl.structure import _Statement, _Slice, _ArrayProxy
 from migen.fhdl.module import Module, FinalizeError
 from migen.fhdl.visit import NodeTransformer
 from migen.fhdl.bitcontainer import value_bits_sign
@@ -16,12 +16,12 @@ class AnonymousState:
 
 # do not use namedtuple here as it inherits tuple
 # and the latter is used elsewhere in FHDL
-class NextState:
+class NextState(_Statement):
     def __init__(self, state):
         self.state = state
 
 
-class NextValue:
+class NextValue(_Statement):
     def __init__(self, target, value):
         self.target = target
         self.value = value