From be063dba14a4d2848cada8fd5c418e06e7c8525c Mon Sep 17 00:00:00 2001
From: Sebastian Junges <sebastian.junges@rwth-aachen.de>
Date: Fri, 24 Jan 2020 18:23:32 +0100
Subject: [PATCH] POMDPs are now always built with choice labelling and choice
 indices

---
 src/storm-cli-utilities/model-handling.h | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/storm-cli-utilities/model-handling.h b/src/storm-cli-utilities/model-handling.h
index a70ddf981..982dd0092 100644
--- a/src/storm-cli-utilities/model-handling.h
+++ b/src/storm-cli-utilities/model-handling.h
@@ -274,8 +274,13 @@ namespace storm {
                 if (counterexampleGeneratorSettings.isCounterexampleSet()) {
                     buildChoiceOrigins = counterexampleGeneratorSettings.isMinimalCommandSetGenerationSet();
                 }
+                options.setBuildChoiceOrigins(buildChoiceOrigins);
             }
-            options.setBuildChoiceOrigins(buildChoiceOrigins);
+            if (input.model->getModelType() == storm::storage::SymbolicModelDescription::ModelType::POMDP) {
+                options.setBuildChoiceOrigins(true);
+                options.setBuildChoiceLabels(true);
+            }
+
             options.setAddOutOfBoundsState(buildSettings.isBuildOutOfBoundsStateSet());
             if (buildSettings.isBuildFullModelSet()) {
                 options.clearTerminalStates();