From adea5713f3d6711885108e359813b4a62ffee98f Mon Sep 17 00:00:00 2001
From: bculkin2442 <bjculkin@mix.wvu.edu>
Date: Sun, 3 Apr 2016 20:36:08 -0400
Subject: Code maintenance and fixes

---
 .../main/java/bjc/dicelang/ast/DiceASTReferenceChecker.java  | 12 ++++--------
 1 file changed, 4 insertions(+), 8 deletions(-)

(limited to 'dice-lang/src/main/java/bjc/dicelang/ast/DiceASTReferenceChecker.java')

diff --git a/dice-lang/src/main/java/bjc/dicelang/ast/DiceASTReferenceChecker.java b/dice-lang/src/main/java/bjc/dicelang/ast/DiceASTReferenceChecker.java
index a74d61e..09d3d13 100644
--- a/dice-lang/src/main/java/bjc/dicelang/ast/DiceASTReferenceChecker.java
+++ b/dice-lang/src/main/java/bjc/dicelang/ast/DiceASTReferenceChecker.java
@@ -2,7 +2,7 @@ package bjc.dicelang.ast;
 
 import java.util.function.Consumer;
 
-import bjc.utils.data.GenHolder;
+import bjc.utils.data.IHolder;
 
 /**
  * Check if the specified node references a particular variable
@@ -15,7 +15,7 @@ public final class DiceASTReferenceChecker
 	/**
 	 * This is true if the specified node references the set variable
 	 */
-	private GenHolder<Boolean>	referencesVariable;
+	private IHolder<Boolean>	referencesVariable;
 
 	private String				varName;
 
@@ -27,7 +27,7 @@ public final class DiceASTReferenceChecker
 	 * @param varName
 	 *            The variable to check for references in
 	 */
-	public DiceASTReferenceChecker(GenHolder<Boolean> referencesVar,
+	public DiceASTReferenceChecker(IHolder<Boolean> referencesVar,
 			String varName) {
 		this.referencesVariable = referencesVar;
 		this.varName = varName;
@@ -35,11 +35,7 @@ public final class DiceASTReferenceChecker
 
 	@Override
 	public void accept(IDiceASTNode astNode) {
-		if (!referencesVariable.unwrap(bool -> bool)) {
-			if (isDirectReference(astNode)) {
-				referencesVariable.transform((bool) -> false);
-			}
-		}
+		referencesVariable.transform((bool) -> isDirectReference(astNode));
 	}
 
 	/**
-- 
cgit v1.2.3