1 files changed, 4 insertions, 4 deletions
diff --git a/src/main/java/bjc/rgens/parser/elements/RuleVariableCaseElement.java b/src/main/java/bjc/rgens/parser/elements/RuleVariableCaseElement.java
index fa1783f..cbd8ce1 100644
--- a/src/main/java/bjc/rgens/parser/elements/RuleVariableCaseElement.java
+++ b/src/main/java/bjc/rgens/parser/elements/RuleVariableCaseElement.java
@@ -18,17 +18,17 @@ public class RuleVariableCaseElement extends VariableCaseElement {
 	}
 
 	public void generate(GenerationState state) {
-		IPair<RGrammar, Rule> par = state.findRule(varDef, true);
+		Rule rl = state.findRule(varDef, true);
 
-		if(par == null) {
+		if(rl == null) {
 			throw new GrammarException("Can't create variable referencing non-existent rule " + varDef);
 		}
 		
 		if(exhaust) {
-			par = new Pair<>(par.getLeft(), par.getRight().exhaust());
+			rl = rl.exhaust();
 		}
 
-		state.rlVars.put(varName, par);
+		state.rlVars.put(varName, rl);
 
 		if(exhaust) {
 			System.err.printf("\t\tFINE: Defined exhausted rulevar '%s' ('%s')\n", varName, varDef);