1 files changed, 5 insertions, 4 deletions
diff --git a/src/test/java/org/zwobble/couscous/tests/frontends/java/FreeVariablesTests.java b/src/test/java/org/zwobble/couscous/tests/frontends/java/FreeVariablesTests.java
index afded63..7cf349d 100644
--- a/src/test/java/org/zwobble/couscous/tests/frontends/java/FreeVariablesTests.java
+++ b/src/test/java/org/zwobble/couscous/tests/frontends/java/FreeVariablesTests.java
@@ -1,6 +1,7 @@
 package org.zwobble.couscous.tests.frontends.java;
 
 import org.junit.Test;
+import org.zwobble.couscous.ast.ReferenceNode;
 import org.zwobble.couscous.ast.VariableDeclaration;
 import org.zwobble.couscous.tests.TestIds;
 import org.zwobble.couscous.values.StringValue;
@@ -21,14 +22,14 @@ public class FreeVariablesTests {
     @Test
     public void referenceIsFreeIfThereIsNoAssociatedDeclaration() {
         VariableDeclaration declaration = var(TestIds.ANY_ID, "[name]", StringValue.REF);
-        List<VariableDeclaration> freeVariables = findFreeVariables(asList(reference(declaration)));
-        assertEquals(asList(declaration), freeVariables);
+        List<ReferenceNode> freeVariables = findFreeVariables(asList(reference(declaration)));
+        assertEquals(asList(reference(declaration)), freeVariables);
     }
 
     @Test
     public void referenceIsNotFreeIfIsDeclaredAsArgument() {
         VariableDeclaration declaration = var(TestIds.ANY_ID, "[name]", StringValue.REF);
-        List<VariableDeclaration> freeVariables = findFreeVariables(
+        List<ReferenceNode> freeVariables = findFreeVariables(
             asList(formalArg(declaration), reference(declaration)));
         assertEquals(emptyList(), freeVariables);
     }
@@ -36,7 +37,7 @@ public class FreeVariablesTests {
     @Test
     public void referenceIsNotFreeIfIsReferenceToLocalDefinedInBody() {
         VariableDeclaration declaration = var(TestIds.ANY_ID, "[name]", StringValue.REF);
-        List<VariableDeclaration> freeVariables = findFreeVariables(
+        List<ReferenceNode> freeVariables = findFreeVariables(
             asList(
                 localVariableDeclaration(declaration, literal("[value]")),
                 reference(declaration)));