diff --git a/src/storm/generator/JaniNextStateGenerator.cpp b/src/storm/generator/JaniNextStateGenerator.cpp
index 62db34455..883abecad 100644
--- a/src/storm/generator/JaniNextStateGenerator.cpp
+++ b/src/storm/generator/JaniNextStateGenerator.cpp
@@ -185,7 +185,7 @@ namespace storm {
             std::vector<StateType> initialStateIndices;
             while (solver->check() == storm::solver::SmtSolver::CheckResult::Sat) {
                 // Create fresh state.
-                CompressedState initialState(this->variableInformation.getTotalBitOffset());
+                CompressedState initialState(this->variableInformation.getTotalBitOffset(true));
                 
                 // Read variable assignment from the solution of the solver. Also, create an expression we can use to
                 // prevent the variable assignment from being enumerated again.
diff --git a/src/storm/generator/PrismNextStateGenerator.cpp b/src/storm/generator/PrismNextStateGenerator.cpp
index 61fe1e80f..9518b2d34 100644
--- a/src/storm/generator/PrismNextStateGenerator.cpp
+++ b/src/storm/generator/PrismNextStateGenerator.cpp
@@ -150,7 +150,7 @@ namespace storm {
             std::vector<StateType> initialStateIndices;
             while (solver->check() == storm::solver::SmtSolver::CheckResult::Sat) {
                 // Create fresh state.
-                CompressedState initialState(this->variableInformation.getTotalBitOffset());
+                CompressedState initialState(this->variableInformation.getTotalBitOffset(true));
                 
                 // Read variable assignment from the solution of the solver. Also, create an expression we can use to
                 // prevent the variable assignment from being enumerated again.