From d2e6f9ed4f4060516e397160af3a15f7a22b4784 Mon Sep 17 00:00:00 2001
From: Sam Vervaeck <samvv@pm.me>
Date: Wed, 31 Aug 2022 13:54:47 +0200
Subject: [PATCH] Fix type-checking return statements

---
 src/checker.ts | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/checker.ts b/src/checker.ts
index c8f07a2f7..f84ab6042 100644
--- a/src/checker.ts
+++ b/src/checker.ts
@@ -524,9 +524,11 @@ export class Checker {
             }
             case SyntaxKind.BlockBody:
             {
+              this.returnTypes.push(returnType);
               for (const element of node.body.elements) {
                 this.infer(element);
               }
+              this.returnTypes.pop();
               break;
             }
           }