diff --git a/crates/ide_assists/src/handlers/generate_constant.rs b/crates/ide_assists/src/handlers/generate_constant.rs index 2e3e3be48d1..a296f1cefcb 100644 --- a/crates/ide_assists/src/handlers/generate_constant.rs +++ b/crates/ide_assists/src/handlers/generate_constant.rs @@ -31,10 +31,9 @@ use syntax::{ // ``` pub(crate) fn generate_constant(acc: &mut Assists, ctx: &AssistContext) -> Option<()> { - let statement = ctx.find_node_at_offset::()?; - let arg_list = ctx.find_node_at_offset::()?; - let expr = ctx.find_node_at_offset::()?; let constant_token = ctx.find_node_at_offset::()?; + let expr = constant_token.syntax().ancestors().find_map(ast::Expr::cast)?; + let statement = expr.syntax().ancestors().find_map(ast::Stmt::cast)?; let ty = ctx.sema.type_of_expr(&expr)?; let scope = ctx.sema.scope(statement.syntax()); let module = scope.module()?;