diff --git a/ReactAndroid/src/main/java/com/facebook/react/uimanager/ReactShadowNode.java b/ReactAndroid/src/main/java/com/facebook/react/uimanager/ReactShadowNode.java
index a3ce58bf3..4357e9157 100644
--- a/ReactAndroid/src/main/java/com/facebook/react/uimanager/ReactShadowNode.java
+++ b/ReactAndroid/src/main/java/com/facebook/react/uimanager/ReactShadowNode.java
@@ -22,6 +22,7 @@ import com.facebook.yoga.YogaConstants;
 import com.facebook.yoga.YogaDirection;
 import com.facebook.yoga.YogaFlexDirection;
 import com.facebook.yoga.YogaJustify;
+import com.facebook.yoga.YogaBaselineFunction;
 import com.facebook.yoga.YogaMeasureFunction;
 import com.facebook.yoga.YogaNode;
 import com.facebook.yoga.YogaOverflow;
@@ -782,6 +783,10 @@ public class ReactShadowNode {
     mShouldNotifyOnLayout = shouldNotifyOnLayout;
   }
 
+  public void setBaselineFunction(YogaBaselineFunction baselineFunction) {
+    mYogaNode.setBaselineFunction(baselineFunction);
+  }
+
   public void setMeasureFunction(YogaMeasureFunction measureFunction) {
     if ((measureFunction == null ^ mYogaNode.isMeasureDefined()) &&
         getChildCount() != 0) {