making stuff work with ReducedStateSpace