1 files changed, 3 insertions, 3 deletions
diff --git a/src/test/java/org/zwobble/couscous/tests/JavaToPythonTests.java b/src/test/java/org/zwobble/couscous/tests/JavaToPythonTests.java
index 6b9f6ed..ecc1fc9 100644
--- a/src/test/java/org/zwobble/couscous/tests/JavaToPythonTests.java
+++ b/src/test/java/org/zwobble/couscous/tests/JavaToPythonTests.java
@@ -1,7 +1,7 @@
 package org.zwobble.couscous.tests;
 
 import org.zwobble.couscous.CouscousCompiler;
-import org.zwobble.couscous.ast.TypeName;
+import org.zwobble.couscous.ast.types.ScalarType;
 import org.zwobble.couscous.backends.python.PythonBackend;
 import org.zwobble.couscous.frontends.java.JavaFrontend;
 import org.zwobble.couscous.tests.backends.python.PythonMethodRunner;
@@ -18,10 +18,10 @@ import static org.zwobble.couscous.util.ExtraLists.list;
 public class JavaToPythonTests extends CompilerTests {
     protected PrimitiveValue execProgram(
             Path directory,
-            TypeName type,
+            ScalarType type,
             String methodName,
             List<PrimitiveValue> arguments,
-            TypeName returnType) throws IOException, InterruptedException {
+            ScalarType returnType) throws IOException, InterruptedException {
 
         Path directoryPath = Files.createTempDirectory(null);
         try {