Read return statements without value

1 files changed, 4 insertions, 1 deletions

diff --git a/src/main/java/org/zwobble/couscous/frontends/java/JavaStatementReader.java b/src/main/java/org/zwobble/couscous/frontends/java/JavaStatementReader.java
index e75585f..6b71684 100644
--- a/src/main/java/org/zwobble/couscous/frontends/java/JavaStatementReader.java
+++ b/src/main/java/org/zwobble/couscous/frontends/java/JavaStatementReader.java
@@ -93,7 +93,10 @@ class JavaStatementReader {
     }
 
     private StatementNode readReturnStatement(ReturnStatement statement) {
-        return ReturnNode.returns(readExpression(returnType.get(), statement.getExpression()));
+        ExpressionNode value = statement.getExpression() == null
+            ? LiteralNode.UNIT
+            : readExpression(returnType.get(), statement.getExpression());
+        return ReturnNode.returns(value);
     }
 
     private StatementNode readThrowStatement(ThrowStatement statement) {