diff --git a/flake.nix b/flake.nix
index cf2259e..9bbde25 100644
--- a/flake.nix
+++ b/flake.nix
@@ -33,8 +33,7 @@
           version =
             let
               commit = inputs.self.shortRev or "dirty";
-              date =
-                inputs.self.lastModifiedDate or inputs.self.lastModified or "19700101";
+              date = inputs.self.lastModifiedDate or inputs.self.lastModified or "19700101";
             in
               "${builtins.substring 0 8 date}_${commit}";
           src = inputs.self.sourceInfo;
diff --git a/src/rules/key_value.rs b/src/rules/key_value.rs
index d230c22..908fda0 100644
--- a/src/rules/key_value.rs
+++ b/src/rules/key_value.rs
@@ -4,9 +4,11 @@ pub fn rule(
 ) -> std::collections::LinkedList<crate::builder::Step> {
     let mut steps = std::collections::LinkedList::new();
 
-    let mut children = crate::children::Children::new(build_ctx, node);
+    let mut children = crate::children::Children::new_with_configuration(
+        build_ctx, node, true,
+    );
 
-    let layout = if children.has_comments() {
+    let layout = if children.has_comments() || children.has_newlines() {
         &crate::config::Layout::Tall
     } else {
         build_ctx.config.layout()
@@ -24,13 +26,17 @@ pub fn rule(
     }
 
     // /**/
-    children.drain_comments(|text| {
-        steps.push_back(crate::builder::Step::NewLine);
-        steps.push_back(crate::builder::Step::Pad);
-        steps.push_back(crate::builder::Step::Comment(text));
+    children.drain_comments_and_newlines(|element| match element {
+        crate::children::DrainCommentOrNewline::Comment(text) => {
+            steps.push_back(crate::builder::Step::NewLine);
+            steps.push_back(crate::builder::Step::Pad);
+            steps.push_back(crate::builder::Step::Comment(text));
+        }
+        crate::children::DrainCommentOrNewline::Newline(_) => {}
     });
 
-    if let rnix::SyntaxKind::TOKEN_COMMENT =
+    if let rnix::SyntaxKind::TOKEN_COMMENT
+    | rnix::SyntaxKind::TOKEN_WHITESPACE =
         children.peek_prev().unwrap().element.kind()
     {
         steps.push_back(crate::builder::Step::NewLine);
@@ -107,10 +113,13 @@ pub fn rule(
     }
 
     // /**/
-    children.drain_comments(|text| {
-        steps.push_back(crate::builder::Step::Comment(text));
-        steps.push_back(crate::builder::Step::NewLine);
-        steps.push_back(crate::builder::Step::Pad);
+    children.drain_comments_and_newlines(|element| match element {
+        crate::children::DrainCommentOrNewline::Comment(text) => {
+            steps.push_back(crate::builder::Step::Comment(text));
+            steps.push_back(crate::builder::Step::NewLine);
+            steps.push_back(crate::builder::Step::Pad);
+        }
+        crate::children::DrainCommentOrNewline::Newline(_) => {}
     });
 
     // b
@@ -125,13 +134,17 @@ pub fn rule(
     }
 
     // /**/
-    children.drain_comments(|text| {
-        steps.push_back(crate::builder::Step::NewLine);
-        steps.push_back(crate::builder::Step::Pad);
-        steps.push_back(crate::builder::Step::Comment(text));
+    children.drain_comments_and_newlines(|element| match element {
+        crate::children::DrainCommentOrNewline::Comment(text) => {
+            steps.push_back(crate::builder::Step::NewLine);
+            steps.push_back(crate::builder::Step::Pad);
+            steps.push_back(crate::builder::Step::Comment(text));
+        }
+        crate::children::DrainCommentOrNewline::Newline(_) => {}
     });
 
-    if let rnix::SyntaxKind::TOKEN_COMMENT =
+    if let rnix::SyntaxKind::TOKEN_COMMENT
+    | rnix::SyntaxKind::TOKEN_WHITESPACE =
         children.peek_prev().unwrap().element.kind()
     {
         steps.push_back(crate::builder::Step::NewLine);
diff --git a/tests/cases/key_value/in b/tests/cases/key_value/in
index 8fb4ca1..0b227d1 100644
--- a/tests/cases/key_value/in
+++ b/tests/cases/key_value/in
@@ -21,6 +21,13 @@ rec /**/ {
   , ...
   }: { };
 
+  a
+  /*b*/
+  =
+    /*c*/
+    1
+    /*d*/
+    ;
 
   p = aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa { }
   a;
diff --git a/tests/cases/key_value/out b/tests/cases/key_value/out
index c19c974..509e0fd 100644
--- a/tests/cases/key_value/out
+++ b/tests/cases/key_value/out
@@ -101,6 +101,20 @@ rec
     }:
     { };
 
+  a
+  /*
+   b
+   */
+  =
+    /*
+     c
+     */
+    1
+    /*
+     d
+     */
+    ;
+
   p =
     aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa { } a;
 }