Introduce TypedFunctionSignatureNode

1 files changed, 4 insertions, 4 deletions

diff --git a/src/test/java/org/shedlang/compiler/ast/typed/TypedFunctionDefinitionNodeMatcher.java b/src/test/java/org/shedlang/compiler/ast/typed/TypedFunctionDefinitionNodeMatcher.java
index c90e2578..92e0c8e1 100644
--- a/src/test/java/org/shedlang/compiler/ast/typed/TypedFunctionDefinitionNodeMatcher.java
+++ b/src/test/java/org/shedlang/compiler/ast/typed/TypedFunctionDefinitionNodeMatcher.java
@@ -35,7 +35,7 @@ public class TypedFunctionDefinitionNodeMatcher extends MatcherBuilder<TypedName
     }
 
     public TypedFunctionDefinitionNodeMatcher withEffect(Matcher<Optional<? extends TypedTypeLevelExpressionNode<?>>> effect) {
-        return with(has("effect", x -> x.effect(), effect));
+        return with(has("effect", x -> x.signature().effect(), effect));
     }
 
     public TypedFunctionDefinitionNodeMatcher withFreeVariables(Matcher<Iterable<FreeVariable>> freeVariables) {
@@ -47,14 +47,14 @@ public class TypedFunctionDefinitionNodeMatcher extends MatcherBuilder<TypedName
     }
 
     public TypedFunctionDefinitionNodeMatcher withParams(Matcher<TypedParamsNode> params) {
-        return with(has("params", x -> x.params(), params));
+        return with(has("params", x -> x.signature().params(), params));
     }
 
     public TypedFunctionDefinitionNodeMatcher withReturnType(Matcher<TypedTypeLevelExpressionNode<?>> returnType) {
-        return with(has("returnType", x -> x.returnType(), returnType));
+        return with(has("returnType", x -> x.signature().returnType(), returnType));
     }
 
     public TypedFunctionDefinitionNodeMatcher withType(FunctionType type) {
-        return with(has("type", x -> x.type(), equalTo(type)));
+        return with(has("type", x -> x.signature().type(), equalTo(type)));
     }
 }