diff --git a/unqlite.c b/unqlite.c
index 49c0b9d..db4e098 100644
--- a/unqlite.c
+++ b/unqlite.c
@@ -20031,8 +20031,8 @@ static sxi32 GenStateCompileFuncBody(
 	/* Restore the default container */
 	jx9VmSetByteCodeContainer(pGen->pVm, pInstrContainer);
 	/* Leave function block */
-	GenStateLeaveBlock(&(*pGen), 0);
-	if( rc == SXERR_ABORT ){
+	rc = GenStateLeaveBlock(&(*pGen), 0);
+	if( rc != SXRET_OK ){
 		/* Don't worry about freeing memory, everything will be released shortly */
 		return SXERR_ABORT;
 	}