diff --git a/codeflash/languages/java/instrumentation.py b/codeflash/languages/java/instrumentation.py index 01ea24661..644300dd6 100644 --- a/codeflash/languages/java/instrumentation.py +++ b/codeflash/languages/java/instrumentation.py @@ -84,9 +84,10 @@ def _is_inside_lambda(node: Any) -> bool: """Check if a tree-sitter node is inside a lambda_expression.""" current = node.parent while current is not None: - if current.type == "lambda_expression": + t = current.type + if t == "lambda_expression": return True - if current.type == "method_declaration": + if t == "method_declaration": return False current = current.parent return False