diff --git a/packages/babel-parser/src/tokenizer/context.js b/packages/babel-parser/src/tokenizer/context.js
index bf07dba9e5ac..b5945e4086c3 100644
--- a/packages/babel-parser/src/tokenizer/context.js
+++ b/packages/babel-parser/src/tokenizer/context.js
@@ -107,7 +107,7 @@ tt._function.updateContext = tt._class.updateContext = function (prevType) {
     !(prevType === tt._return && this.hasPrecedingLineBreak()) &&
     !(
       (prevType === tt.colon || prevType === tt.braceL) &&
-      this.curContext() === types.b_stat
+      this.curContext() === types.braceStatement
     )
   ) {
     this.state.context.push(types.functionExpression);