diff --git a/prism/src/prism/ModelType.java b/prism/src/prism/ModelType.java
index fd8c6f67..da82c55c 100644
--- a/prism/src/prism/ModelType.java
+++ b/prism/src/prism/ModelType.java
@@ -135,6 +135,21 @@ public enum ModelType {
 		return true;
 	}
 	
+	/**
+	 * Does this model have probabilities or rates?
+	 * @return "Probability" or "Rate"
+	 */
+	public String probabilityOrRate()
+	{
+		switch (this) {
+		case CTMC:
+		case CTMDP:
+			return "Rate";
+		default:
+			return "Probability";
+		}
+	}
+	
 	public static ModelType parseName(String name)
 	{
 		if ("ctmc".equals(name))
diff --git a/prism/src/userinterface/simulator/GUISimulator.java b/prism/src/userinterface/simulator/GUISimulator.java
index 6fab46d1..21ad82ee 100644
--- a/prism/src/userinterface/simulator/GUISimulator.java
+++ b/prism/src/userinterface/simulator/GUISimulator.java
@@ -2077,11 +2077,7 @@ public class GUISimulator extends GUIPlugin implements MouseListener, ListSelect
 				case 0:
 					return "Module/[action]";
 				case 1: {
-					if (parsedModel != null && parsedModel.getModelType() == ModelType.CTMC)
-						return "Rate";
-					else
-						return "Probability";
-
+					return parsedModel == null ? "Probability" : parsedModel.getModelType().probabilityOrRate();
 				}
 				case 2:
 					return "Update";