diff --git a/ReactCommon/fabric/components/view/yoga/YogaLayoutableShadowNode.cpp b/ReactCommon/fabric/components/view/yoga/YogaLayoutableShadowNode.cpp
index f4bd3ca6ddd..0c87c9094ec 100644
--- a/ReactCommon/fabric/components/view/yoga/YogaLayoutableShadowNode.cpp
+++ b/ReactCommon/fabric/components/view/yoga/YogaLayoutableShadowNode.cpp
@@ -245,8 +245,11 @@ void YogaLayoutableShadowNode::layoutTree(
   {
     SystraceSection s("YogaLayoutableShadowNode::YGNodeCalculateLayout");
 
-    YGNodeCalculateLayout(
-        &yogaNode_, availableWidth, availableHeight, YGDirectionInherit);
+    YGNodeCalculateLayoutWithContext(&yogaNode_,
+                                     availableWidth,
+                                     availableHeight,
+                                     YGDirectionInherit,
+                                     &layoutContext);
   }
 
   layout(layoutContext);
@@ -320,7 +323,8 @@ YGSize YogaLayoutableShadowNode::yogaNodeMeasureCallbackConnector(
     float width,
     YGMeasureMode widthMode,
     float height,
-    YGMeasureMode heightMode) {
+    YGMeasureMode heightMode,
+    void *layoutContext) {
   SystraceSection s(
       "YogaLayoutableShadowNode::yogaNodeMeasureCallbackConnector");
 
diff --git a/ReactCommon/fabric/components/view/yoga/YogaLayoutableShadowNode.h b/ReactCommon/fabric/components/view/yoga/YogaLayoutableShadowNode.h
index ceb0de20f6a..303609b7541 100644
--- a/ReactCommon/fabric/components/view/yoga/YogaLayoutableShadowNode.h
+++ b/ReactCommon/fabric/components/view/yoga/YogaLayoutableShadowNode.h
@@ -121,7 +121,8 @@ class YogaLayoutableShadowNode : public LayoutableShadowNode {
       float width,
       YGMeasureMode widthMode,
       float height,
-      YGMeasureMode heightMode);
+      YGMeasureMode heightMode,
+      void *layoutContext);
 };
 
 template <>