Make tests pass again, slightly refactor the `ExplicitStateUpdater` code (but `_stochastic_part` and `_non_stochastic_part` are still very redundant)