2 files changed, 6 insertions, 3 deletions
diff --git a/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFunctionDefinitionTests.java b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFunctionDefinitionTests.java
index 86b38d5c..8a096311 100644
--- a/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFunctionDefinitionTests.java
+++ b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFunctionDefinitionTests.java
@@ -352,7 +352,8 @@ public class TypeCheckerFunctionDefinitionTests {
         UntypedFunctionDefinitionNode untyped,
         TypeCheckerContext context
     ) {
-        var environment = TypeChecker.declareVariables(untyped, EnvironmentBuilder.EMPTY);
+        var declaredVariables = TypeChecker.declaredVariables(untyped, P.vector());
+        var environment = EnvironmentBuilder.EMPTY.addUndefined(declaredVariables);
 
         context = context.enterNamespace(environment.build());
 
diff --git a/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerValTests.java b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerValTests.java
index 6c23f08b..ee9284b2 100644
--- a/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerValTests.java
+++ b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerValTests.java
@@ -149,7 +149,8 @@ public class TypeCheckerValTests {
         UntypedFunctionStatementNode untyped,
         TypeCheckerContext context
     ) {
-        var environment = TypeChecker.declareVariables(untyped, EnvironmentBuilder.EMPTY);
+        var declaredVariables = TypeChecker.declaredVariables(untyped, P.vector());
+        var environment = EnvironmentBuilder.EMPTY.addUndefined(declaredVariables);
 
         context = context.enterNamespace(environment.build());
 
@@ -160,7 +161,8 @@ public class TypeCheckerValTests {
         UntypedNamespaceStatementNode untyped,
         TypeCheckerContext context
     ) {
-        var environment = TypeChecker.declareVariables(untyped, EnvironmentBuilder.EMPTY);
+        var declaredVariables = TypeChecker.declaredVariables(untyped, P.vector());
+        var environment = EnvironmentBuilder.EMPTY.addUndefined(declaredVariables);
 
         context = context.enterNamespace(environment.build());