1 files changed, 8 insertions, 14 deletions
diff --git a/src/test/java/org/zwobble/hobgoblin/compiler/typechecker/TypeCheckerNamespaceTests.java b/src/test/java/org/zwobble/hobgoblin/compiler/typechecker/TypeCheckerNamespaceTests.java
index de0ffec..5a2f7cb 100644
--- a/src/test/java/org/zwobble/hobgoblin/compiler/typechecker/TypeCheckerNamespaceTests.java
+++ b/src/test/java/org/zwobble/hobgoblin/compiler/typechecker/TypeCheckerNamespaceTests.java
@@ -1,12 +1,10 @@
 package org.zwobble.hobgoblin.compiler.typechecker;
 
 import org.junit.jupiter.api.Test;
-import org.zwobble.hobgoblin.compiler.ast.DocComment;
 import org.zwobble.hobgoblin.compiler.ast.typed.TypedNamespaceNode;
 import org.zwobble.hobgoblin.compiler.ast.typed.TypedStructDefinitionNode;
 import org.zwobble.hobgoblin.compiler.ast.untyped.UntypedNamespaceNode;
 import org.zwobble.hobgoblin.compiler.ast.untyped.UntypedStructDefinitionNode;
-import org.zwobble.hobgoblin.compiler.sources.NullSource;
 import org.zwobble.hobgoblin.compiler.types.NamespaceName;
 import org.zwobble.hobgoblin.compiler.types.StructType;
 
@@ -18,18 +16,14 @@ import static org.zwobble.precisely.Matchers.*;
 public class TypeCheckerNamespaceTests {
     @Test
     public void namespaceStatementsAreTypeChecked() {
-        var untyped = new UntypedNamespaceNode(
-            NamespaceName.of("a", "b"),
-            List.of(
-                new UntypedStructDefinitionNode(
-                    "X",
-                    List.of(),
-                    DocComment.EMPTY,
-                    NullSource.INSTANCE
-                )
-            ),
-            NullSource.INSTANCE
-        );
+        var untyped = UntypedNamespaceNode.arbitrary()
+            .withNamespaceName(NamespaceName.of("a", "b"))
+            .withBody(List.of(
+                UntypedStructDefinitionNode.arbitrary()
+                    .withName("X")
+                    .build()
+            ))
+            .build();
         var context = TypeCheckerContextArb.globalContext();
 
         var typed = TypeChecker.typeCheckNamespace(untyped, context);