diff --git a/prism/src/simulator/ModulesFileModelGenerator.java b/prism/src/simulator/ModulesFileModelGenerator.java
index 91c9a5b7..699cac9e 100644
--- a/prism/src/simulator/ModulesFileModelGenerator.java
+++ b/prism/src/simulator/ModulesFileModelGenerator.java
@@ -268,17 +268,17 @@ public class ModulesFileModelGenerator extends DefaultModelGenerator
 	}
 
 	@Override
-	public String getTransitionAction(int index, int offset) throws PrismException
+	public String getTransitionAction(int index) throws PrismException
 	{
-		TransitionList transitions = getTransitionList();
-		int a = transitions.getTransitionModuleOrActionIndex(transitions.getTotalIndexOfTransition(index, offset));
+		int a = getTransitionList().getTransitionModuleOrActionIndex(index);
 		return a < 0 ? null : modulesFile.getSynch(a - 1);
 	}
 
-	//@Override
-	public String getTransitionAction(int index) throws PrismException
+	@Override
+	public String getTransitionAction(int index, int offset) throws PrismException
 	{
-		int a = getTransitionList().getTransitionModuleOrActionIndex(index);
+		TransitionList transitions = getTransitionList();
+		int a = transitions.getTransitionModuleOrActionIndex(transitions.getTotalIndexOfTransition(index, offset));
 		return a < 0 ? null : modulesFile.getSynch(a - 1);
 	}