From a7f9a6e4c63684120ec61ad700c99b31802d1cda Mon Sep 17 00:00:00 2001
From: Sebastian Junges <sebastian.junges@gmail.com>
Date: Fri, 26 Mar 2021 12:43:43 -0700
Subject: [PATCH] use state rewards (upon entry)

---
 src/storm/simulator/PrismProgramSimulator.cpp | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/storm/simulator/PrismProgramSimulator.cpp b/src/storm/simulator/PrismProgramSimulator.cpp
index 0b72f6f58..6f693b3ce 100644
--- a/src/storm/simulator/PrismProgramSimulator.cpp
+++ b/src/storm/simulator/PrismProgramSimulator.cpp
@@ -44,6 +44,9 @@ namespace storm {
             if (behavior.getStateRewards().size() > 0) {
                 STORM_LOG_ASSERT(behavior.getStateRewards().size() == lastActionRewards.size(), "Reward vectors should have same length.");
             }
+            for(uint64_t i = 0; i < behavior.getStateRewards().size(); i++) {
+                lastActionRewards[i] += behavior.getStateRewards()[i];
+            }
             return true;
         }