diff --git a/prism/src/explicit/CounterTransformation.java b/prism/src/explicit/CounterTransformation.java
index 21a1aab0..16555fa9 100644
--- a/prism/src/explicit/CounterTransformation.java
+++ b/prism/src/explicit/CounterTransformation.java
@@ -289,7 +289,7 @@ public class CounterTransformation<M extends Model> implements ModelExpressionTr
 		for (TemporalOperatorBound bound : bounds) {
 			// resolve RewardStruct for reward bounds
 			if (bound.isRewardBound()) {
-				int r = ExpressionReward.getRewardStructIndexByIndexObject(bound.getRewardStructureIndex(), parent.modulesFile, parent.constantValues);
+				int r = ExpressionReward.getRewardStructIndexByIndexObject(bound.getRewardStructureIndex(), parent.rewardGen, parent.constantValues);
 				bound.setResolvedRewardStructIndex(r);
 			}
 		}
@@ -298,7 +298,7 @@ public class CounterTransformation<M extends Model> implements ModelExpressionTr
 
 		for (List<TemporalOperatorBound> groupedBounds : groupedBoundList) {
 			if (groupedBounds.get(0).isRewardBound()) {
-				String rewStructName = parent.modelInfo.getRewardStructNames().get(groupedBounds.get(0).getResolvedRewardStructIndex());
+				String rewStructName = parent.rewardGen.getRewardStructNames().get(groupedBounds.get(0).getResolvedRewardStructIndex());
 				parent.getLog().println("Transform to incorporate counter for reward '" + rewStructName + "' and " + groupedBounds);
 			} else {
 				parent.getLog().println("Transform to incorporate counter for steps "+groupedBounds);