diff --git a/src/storm/storage/jani/Automaton.cpp b/src/storm/storage/jani/Automaton.cpp index 08cec4105..37924c221 100644 --- a/src/storm/storage/jani/Automaton.cpp +++ b/src/storm/storage/jani/Automaton.cpp @@ -444,9 +444,11 @@ namespace storm { for (auto& location : this->getLocations()) { location.substitute(substitution); } - - this->setInitialStatesRestriction(substituteJaniExpression(this->getInitialStatesRestriction(), substitution)); - + + if (hasInitialStatesRestriction()) { + this->setInitialStatesRestriction(substituteJaniExpression(this->getInitialStatesRestriction(), substitution)); + } + edges.substitute(substitution); } void Automaton::registerTemplateEdge(std::shared_ptr const& te) {