aboutsummaryrefslogtreecommitdiff
path: root/src/main/java
diff options
context:
space:
mode:
Diffstat (limited to 'src/main/java')
-rw-r--r--src/main/java/org/shedlang/compiler/typechecker/TypeChecker.java2
1 files changed, 1 insertions, 1 deletions
diff --git a/src/main/java/org/shedlang/compiler/typechecker/TypeChecker.java b/src/main/java/org/shedlang/compiler/typechecker/TypeChecker.java
index dcb18033..049bdee6 100644
--- a/src/main/java/org/shedlang/compiler/typechecker/TypeChecker.java
+++ b/src/main/java/org/shedlang/compiler/typechecker/TypeChecker.java
@@ -283,7 +283,7 @@ public class TypeChecker {
var bodyContext = context.enterFunction(
bodyEnvironment.build(),
typedEffect.map(e -> e.value()).orElse(Effects.NONE),
- Types.UNIT
+ typedReturnType.value()
);
var bodyResult = bodyContext.mapWithContext(