Rework style chain access

1 files changed, 3 insertions, 4 deletions

diff --git a/library/src/layout/stack.rs b/library/src/layout/stack.rs
index 430af715..f4e4ab2c 100644
--- a/library/src/layout/stack.rs
+++ b/library/src/layout/stack.rs
@@ -201,10 +201,9 @@ impl<'a> StackLayouter<'a> {
 
         // Block-axis alignment of the `AlignNode` is respected
         // by the stack node.
-        let aligns = if let Some(styled) = block.to::<StyledNode>() {
-            styles.chain(&styled.map()).get(AlignNode::ALIGNMENT)
-        } else {
-            styles.get(AlignNode::ALIGNMENT)
+        let aligns = match block.to::<StyledNode>() {
+            Some(styled) => AlignNode::alignment_in(styles.chain(&styled.map())),
+            None => AlignNode::alignment_in(styles),
         };
 
         let aligns = aligns.resolve(styles);