diff --git a/src/widgets/toplevels/toplevel_layout/row_col_toplevel_layout.rs b/src/widgets/toplevels/toplevel_layout/row_col_toplevel_layout.rs
index e3e116c..ee51f53 100644
--- a/src/widgets/toplevels/toplevel_layout/row_col_toplevel_layout.rs
+++ b/src/widgets/toplevels/toplevel_layout/row_col_toplevel_layout.rs
@@ -21,10 +21,15 @@ impl RowColToplevelLayout {
 
 impl ToplevelLayout for RowColToplevelLayout {
     fn size(&self) -> Size<Length> {
-        Size {
-            width: Length::Fill,
-            // TODO Make depend on orientation or drop that option
-            height: Length::Shrink,
+        match self.axis {
+            Axis::Horizontal => Size {
+                width: Length::Fill,
+                height: Length::Shrink,
+            },
+            Axis::Vertical => Size {
+                width: Length::Shrink,
+                height: Length::Fill,
+            },
         }
     }