diff --git a/src/rules/key_value.rs b/src/rules/key_value.rs
index 3dd6e06..5b49ac9 100644
--- a/src/rules/key_value.rs
+++ b/src/rules/key_value.rs
@@ -47,6 +47,7 @@ pub fn rule(
         crate::config::Layout::Tall => {
             if let rnix::SyntaxKind::NODE_APPLY
             | rnix::SyntaxKind::NODE_ATTR_SET
+            | rnix::SyntaxKind::NODE_LAMBDA
             | rnix::SyntaxKind::NODE_LIST
             | rnix::SyntaxKind::NODE_PAREN
             | rnix::SyntaxKind::NODE_STRING =
diff --git a/tests/cases/key_value/in b/tests/cases/key_value/in
index 5146a9d..166149e 100644
--- a/tests/cases/key_value/in
+++ b/tests/cases/key_value/in
@@ -7,4 +7,8 @@
   f = {a/*b*/=     1/*d*/;};
   h = {a/*b*/=/*c*/1     ;};
   i = {a/*b*/=/*c*/1/*d*/;};
+  j = a:       {    b = 1           ;};
+  k = a:       {    b = 1;     c = 2;};
+  l = a: /*b*/ {    b = 1           ;};
+  m = a: /*b*/ {    b = 1;     c = 2;};
 }
diff --git a/tests/cases/key_value/out b/tests/cases/key_value/out
index 62004b2..519fdd9 100644
--- a/tests/cases/key_value/out
+++ b/tests/cases/key_value/out
@@ -72,4 +72,22 @@
        */
       ;
   };
+  j = a: { b = 1; };
+  k = a: {
+    b = 1;
+    c = 2;
+  };
+  l = a:
+  /*
+   b
+   */
+  { b = 1; };
+  m = a:
+  /*
+   b
+   */
+  {
+    b = 1;
+    c = 2;
+  };
 }