diff --git a/src/check_ast.rs b/src/check_ast.rs
index 025174ed71..be4df3f4ef 100644
--- a/src/check_ast.rs
+++ b/src/check_ast.rs
@@ -457,7 +457,7 @@ where
                     ));
                 }
             }
-            ExprKind::Yield { .. } | ExprKind::YieldFrom { .. } => {
+            ExprKind::Yield { .. } | ExprKind::YieldFrom { .. } | ExprKind::Await { .. } => {
                 let scope =
                     &self.scopes[*(self.scope_stack.last().expect("No current scope found."))];
                 if self