1 files changed, 2 insertions, 6 deletions
diff --git a/src/main/java/org/shedlang/compiler/typechecker/TypeChecker.java b/src/main/java/org/shedlang/compiler/typechecker/TypeChecker.java
index 3cc5b541..1d1e73a9 100644
--- a/src/main/java/org/shedlang/compiler/typechecker/TypeChecker.java
+++ b/src/main/java/org/shedlang/compiler/typechecker/TypeChecker.java
@@ -451,12 +451,8 @@ public class TypeChecker {
     ) {
         var typedParams = typeCheckParams(untyped.params(), context);
 
-        Optional<TypedTypeLevelExpressionNode<Effect>> typedEffect;
-        if (untyped.effect().isPresent()) {
-            typedEffect = Optional.of(typeCheckEffect(untyped.effect().get(), context));
-        } else {
-            typedEffect = Optional.empty();
-        }
+        var typedEffect = untyped.effect()
+            .map(untypedEffect -> typeCheckEffect(untyped.effect().get(), context));
 
         var typedReturnType = typeCheckTypeLevelExpression(TypeSets.META_TYPE, untyped.returnType(), context);