From f5a9ebe2f7da3677fd5c99a15c3c03d3a58b2357 Mon Sep 17 00:00:00 2001
From: dehnert <dehnert@cs.rwth-aachen.de>
Date: Sun, 1 Feb 2015 23:52:40 +0100
Subject: [PATCH] And another minor bug.

Former-commit-id: af2e9d05ee3731dbf23b3c52b22d2ba92094dd8a
---
 src/stormParametric.cpp | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/stormParametric.cpp b/src/stormParametric.cpp
index 049f638f1..137a2b285 100644
--- a/src/stormParametric.cpp
+++ b/src/stormParametric.cpp
@@ -154,6 +154,11 @@ void check() {
     
     std::shared_ptr<storm::models::AbstractModel<ValueType>> model = storm::builder::ExplicitPrismModelBuilder<ValueType>::translateProgram(program, options);
     
+    // Convert the transition rewards to state rewards if necessary.
+    if (model->hasTransitionRewards()) {
+        model->convertTransitionRewardsToStateRewards();
+    }
+    
     model->printModelInformationToStream(std::cout);
     
     // Program Translation Time Measurement, End