Small style comment: perhaps also import `State.*` here, if that's possible (perhaps not, since it's declared in this class)?