diff --git a/prism/src/strat/MDStrategy.java b/prism/src/strat/MDStrategy.java
index 648a140b..72d66aaf 100644
--- a/prism/src/strat/MDStrategy.java
+++ b/prism/src/strat/MDStrategy.java
@@ -70,7 +70,7 @@ public abstract class MDStrategy implements Strategy
 	}
 	
 	@Override
-	public void update(int action, int s)
+	public void update(Object action, int s)
 	{
 		currentState = s;
 	}
diff --git a/prism/src/strat/Strategy.java b/prism/src/strat/Strategy.java
index c7b6f671..aa73cc15 100644
--- a/prism/src/strat/Strategy.java
+++ b/prism/src/strat/Strategy.java
@@ -55,7 +55,7 @@ public interface Strategy
 	 * @param action The action taken in the previous state of the model
 	 * @param s The new state of the model
 	 */
-	public void update(int action, int s);
+	public void update(Object action, int s);
 	
 	/**
 	 * Get the action chosen by the strategy in the current state (assuming it is deterministic).