1 files changed, 5 insertions, 2 deletions
diff --git a/src/test/java/org/zwobble/couscous/tests/MethodRunner.java b/src/test/java/org/zwobble/couscous/tests/MethodRunner.java
index 8fde77f..cf7b884 100644
--- a/src/test/java/org/zwobble/couscous/tests/MethodRunner.java
+++ b/src/test/java/org/zwobble/couscous/tests/MethodRunner.java
@@ -3,6 +3,8 @@ package org.zwobble.couscous.tests;
 import com.google.common.collect.ImmutableList;
 import com.google.common.collect.Iterables;
 import org.zwobble.couscous.ast.*;
+import org.zwobble.couscous.ast.types.ScalarType;
+import org.zwobble.couscous.ast.types.Type;
 import org.zwobble.couscous.values.PrimitiveValue;
 
 import java.util.List;
@@ -12,9 +14,10 @@ import static org.zwobble.couscous.util.ExtraLists.list;
 public interface MethodRunner {
     PrimitiveValue runMethod(
         List<TypeNode> classNodes,
-        TypeName className,
+        ScalarType className,
         String methodName,
-        List<PrimitiveValue> arguments, TypeName returnType);
+        List<PrimitiveValue> arguments,
+        Type returnType);
 
     default PrimitiveValue evalExpression(List<TypeNode> classes, ExpressionNode expression) {
         ClassNode programNode = ClassNode.builder("Program")