diff --git a/src/storm-pomdp/transformer/KnownProbabilityTransformer.cpp b/src/storm-pomdp/transformer/KnownProbabilityTransformer.cpp index c5d225d02..45016e515 100644 --- a/src/storm-pomdp/transformer/KnownProbabilityTransformer.cpp +++ b/src/storm-pomdp/transformer/KnownProbabilityTransformer.cpp @@ -16,7 +16,9 @@ namespace storm { std::map stateMap; std::map observationMap; - storm::models::sparse::StateLabeling newLabeling(pomdp.getNumberOfStates() - prob0States.getNumberOfSetBits() - prob1States.getNumberOfSetBits() + 2); + uint64_t nrNewStates = prob0States.empty() ? 1 : 2; + + storm::models::sparse::StateLabeling newLabeling(pomdp.getNumberOfStates() - prob0States.getNumberOfSetBits() - prob1States.getNumberOfSetBits() + nrNewStates); std::vector newObservations;