diff --git a/yoga/Yoga.cpp b/yoga/Yoga.cpp
index 91e09c15..b49924b2 100644
--- a/yoga/Yoga.cpp
+++ b/yoga/Yoga.cpp
@@ -2127,7 +2127,7 @@ static float YGDistributeFreeSpaceSecondPass(
                          currentRelativeChild,
                          mainAxis,
                          currentRelativeChild->getLayout().computedFlexBasis,
-                         mainAxisownerSize)
+                         availableInnerMainDim)
                          .unwrap();
     float updatedMainSize = childFlexBasis;
 
@@ -2313,7 +2313,7 @@ static void YGDistributeFreeSpaceFirstPass(
             currentRelativeChild,
             mainAxis,
             currentRelativeChild->getLayout().computedFlexBasis,
-            mainAxisownerSize)
+            availableInnerMainDim)
             .unwrap();
 
     if (collectedFlexItemsValues.remainingFreeSpace < 0) {