1 files changed, 4 insertions, 3 deletions
diff --git a/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerNamespaceTests.java b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerNamespaceTests.java
index bbc4947c..b9b6679c 100644
--- a/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerNamespaceTests.java
+++ b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerNamespaceTests.java
@@ -5,6 +5,7 @@ import org.shedlang.compiler.ast.untyped.Untyped;
 import org.shedlang.compiler.ast.untyped.UntypedFunctionDefinitionNode;
 import org.shedlang.compiler.types.FunctionType;
 import org.shedlang.compiler.types.NamespaceName;
+import org.shedlang.compiler.types.NamespaceTypeInfo;
 import org.shedlang.compiler.types.Types;
 import org.shedlang.compiler.util.P;
 
@@ -74,10 +75,10 @@ public class TypeCheckerNamespaceTests {
         var result = TypeChecker.typeCheckNamespace(untyped, context);
 
         assertThat(
-            result.context().namespaceFields().get(namespaceName),
-            equalTo(P.orderedMap(
+            result.context().namespaceTypeInfo().get(namespaceName),
+            equalTo(new NamespaceTypeInfo(P.orderedMap(
                 Map.entry("f", new FunctionType(List.of(), List.of(), Types.UNIT))
-            ))
+            )))
         );
     }
 }