1 files changed, 2 insertions, 12 deletions
diff --git a/examples/02-sum/output/java/src/test/java/org/zwobble/example/ShapesTests.java b/examples/02-sum/output/java/src/test/java/org/zwobble/example/ShapesTests.java
index 2554ca5..893a544 100644
--- a/examples/02-sum/output/java/src/test/java/org/zwobble/example/ShapesTests.java
+++ b/examples/02-sum/output/java/src/test/java/org/zwobble/example/ShapesTests.java
@@ -10,24 +10,14 @@ public class ShapesTests {
     @Test
     public void rectangle() {
         Shape rectangle = new Rectangle(10, 25);
-        int area = area(rectangle);
+        int area = Shapes.area(rectangle);
         assertEquals(area, 250);
     }
 
     @Test
     public void triangle() {
         Shape triangle = new Triangle(10, 25);
-        int area = area(triangle);
+        int area = Shapes.area(triangle);
         assertEquals(area, 125);
     }
-
-    private static int area(Shape shape) {
-        return switch (shape) {
-            case Rectangle(var width, var height) ->
-                width * height;
-
-            case Triangle(var width, var height) ->
-                width * height / 2;
-        };
-    }
 }