Use orElseThrow() instead of get()

1 files changed, 3 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 1d1e73a9..a5996ef7 100644
--- a/src/main/java/org/shedlang/compiler/typechecker/TypeChecker.java
+++ b/src/main/java/org/shedlang/compiler/typechecker/TypeChecker.java
@@ -44,13 +44,10 @@ public class TypeChecker {
     }
 
     private static Type compileTimeLookup(String name, Source source, TypeCheckerContext context) {
-        var variable = context.compileTimeLookup(name);
+        var variable = context.compileTimeLookup(name)
+            .orElseThrow(() -> new UnknownVariableError(name, source));
 
-        if (variable.isEmpty()) {
-            throw new UnknownVariableError(name, source);
-        }
-
-        return variable.get().type()
+        return variable.type()
             .orElseThrow(() -> new ReferenceBeforeDefinitionError(name, source));
     }