From 814498ada2c07005a9b1bf957d85bfdf09a8fd84 Mon Sep 17 00:00:00 2001
From: Stefan Pranger <stefan.pranger@student.tugraz.at>
Date: Tue, 16 Mar 2021 13:19:37 +0100
Subject: [PATCH] fix output of prob choice labels in scheduler

---
 src/storm/storage/PostScheduler.cpp | 2 +-
 src/storm/storage/Scheduler.cpp     | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/storm/storage/PostScheduler.cpp b/src/storm/storage/PostScheduler.cpp
index 2208abb36..74bb2983c 100644
--- a/src/storm/storage/PostScheduler.cpp
+++ b/src/storm/storage/PostScheduler.cpp
@@ -144,7 +144,7 @@ namespace storm {
                                     stateString << choiceProbPair.first;
                                 }
                                 if (choiceLabelsGiven) {
-                                    auto choiceLabels = model->getChoiceLabeling().getLabelsOfChoice(model->getTransitionMatrix().getRowGroupIndices()[state] + choice.getDeterministicChoice());
+                                    auto choiceLabels = model->getChoiceLabeling().getLabelsOfChoice(model->getTransitionMatrix().getRowGroupIndices()[state] + choiceProbPair.first);
                                     stateString << " {" << boost::join(choiceLabels, ", ") << "}";
                                 }
                                 stateString << ")";
diff --git a/src/storm/storage/Scheduler.cpp b/src/storm/storage/Scheduler.cpp
index 72b87234a..c53d1f015 100644
--- a/src/storm/storage/Scheduler.cpp
+++ b/src/storm/storage/Scheduler.cpp
@@ -204,7 +204,7 @@ namespace storm {
                                     stateString << choiceProbPair.first;
                                 }
                                 if (choiceLabelsGiven) {
-                                    auto choiceLabels = model->getChoiceLabeling().getLabelsOfChoice(model->getTransitionMatrix().getRowGroupIndices()[state] + choice.getDeterministicChoice());
+                                    auto choiceLabels = model->getChoiceLabeling().getLabelsOfChoice(model->getTransitionMatrix().getRowGroupIndices()[state] + choiceProbPair.first);
                                     stateString << " {" << boost::join(choiceLabels, ", ") << "}";
                                 }
                                 stateString << ")";