From 610d472b66600146fbdf990e4e40aac1ab108a29 Mon Sep 17 00:00:00 2001
From: Brian Fiete <bfiete@gmail.com>
Date: Wed, 19 Feb 2020 14:02:49 -0800
Subject: [PATCH] Fixed new defer warning

---
 IDEHelper/Compiler/BfModule.cpp | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/IDEHelper/Compiler/BfModule.cpp b/IDEHelper/Compiler/BfModule.cpp
index cd76a011..7c3eb17a 100644
--- a/IDEHelper/Compiler/BfModule.cpp
+++ b/IDEHelper/Compiler/BfModule.cpp
@@ -18154,6 +18154,12 @@ BfModuleMethodInstance BfModule::GetLocalMethodInstance(BfLocalMethod* localMeth
 	methodState.mIRFunction = declMethodState->mIRFunction;
 	methodState.mDeferredLocalAssignData = &deferredLocalAssignData;
 
+	if (auto blockBody = BfNodeDynCast<BfBlock>(body))
+	{
+		methodState.mCurScope->mAstBlock = blockBody;
+		methodState.mCurScope->mCloseNode = blockBody->mCloseBrace;
+	}
+
 	mBfIRBuilder->SetInsertPoint(methodState.mIREntryBlock);	
 
 	BfClosureState closureState;