1 files changed, 5 insertions, 4 deletions
diff --git a/src/main/java/org/zwobble/couscous/backends/python/PythonCompiler.java b/src/main/java/org/zwobble/couscous/backends/python/PythonCompiler.java
index a54f037..67e24cc 100644
--- a/src/main/java/org/zwobble/couscous/backends/python/PythonCompiler.java
+++ b/src/main/java/org/zwobble/couscous/backends/python/PythonCompiler.java
@@ -5,6 +5,7 @@ import java.nio.file.Files;
 import java.nio.file.Path;
 import java.util.List;
 
+import org.zwobble.couscous.ast.ClassName;
 import org.zwobble.couscous.ast.ClassNode;
 
 import static java.util.Arrays.asList;
@@ -25,15 +26,15 @@ public class PythonCompiler {
         for (val classNode : classes) {
             compileClass(classNode);
         }
-        writeClass("java.lang.Integer",
+        writeClass(ClassName.of("java.lang.Integer"),
             "class Integer(object):\n" +
             "    def parseInt(value):\n" +
             "        return int(value)"
         );
     }
 
-    private Path pathForClass(String className) {
-        return root.resolve(className.replace(".", File.separator) + ".py");
+    private Path pathForClass(ClassName className) {
+        return root.resolve(className.getQualifiedName().replace(".", File.separator) + ".py");
     }
 
     private void compileClass(ClassNode classNode) {
@@ -41,7 +42,7 @@ public class PythonCompiler {
     }
 
     @SneakyThrows
-    private void writeClass(String name, String contents) {
+    private void writeClass(ClassName name, String contents) {
         val path = pathForClass(name);
         Files.createDirectories(path.getParent());
         Files.write(path, asList(contents));