GrStatement (GrStatement branch) { if (branch instanceof GrBlockStatement block) { final GrStatement[] statements = block.getBlock().getStatements(); if (statements.length == 1) { return statements[0]; } else { return block; } } else { return branch; } }