diff --git a/crates/ruff_python_formatter/src/statement/suite.rs b/crates/ruff_python_formatter/src/statement/suite.rs index 7623ef9449e90b..f5f874ec10f065 100644 --- a/crates/ruff_python_formatter/src/statement/suite.rs +++ b/crates/ruff_python_formatter/src/statement/suite.rs @@ -138,7 +138,13 @@ impl FormatRule> for FormatSuite { SuiteChildStatement::Other(first) } } - SuiteKind::TopLevel => SuiteChildStatement::Other(first), + SuiteKind::TopLevel => { + if let Some(docstring) = DocstringStmt::try_from_statement(first, self.kind) { + SuiteChildStatement::Docstring(docstring) + } else { + SuiteChildStatement::Other(first) + } + } }; let first_comments = comments.leading_dangling_trailing(first);