diff --git a/csharp/Facebook.Yoga/YogaNode.cs b/csharp/Facebook.Yoga/YogaNode.cs
index 6ea40e42..ef7f3bf9 100644
--- a/csharp/Facebook.Yoga/YogaNode.cs
+++ b/csharp/Facebook.Yoga/YogaNode.cs
@@ -64,8 +64,8 @@ namespace Facebook.Yoga
         private BaselineFunction _baselineFunction;
         private object _data;
 #if (UNITY_IOS && !UNITY_EDITOR) || __IOS__
-        private static YogaMeasureFunc _managedMeasure = MeasureInternalIOS;
-        private static YogaBaselineFunc _managedBaseline = BaselineInternalIOS;
+        private static YogaMeasureFunc _managedMeasure;
+        private static YogaBaselineFunc _managedBaseline;
 #else
         private YogaMeasureFunc _managedMeasure;
         private YogaBaselineFunc _managedBaseline;
@@ -617,11 +617,16 @@ namespace Facebook.Yoga
             if (measureFunction != null)
             {
 #if (UNITY_IOS && !UNITY_EDITOR) || __IOS__
+                _managedMeasure = MeasureInternalIOS;
                 _ygNode.SetContext(this);
 #else
                 _managedMeasure = MeasureInternal;
 #endif
             }
+            else
+            {
+                _managedMeasure = null;
+            }
             Native.YGNodeSetMeasureFunc(_ygNode, _managedMeasure);
         }
 
@@ -631,11 +636,16 @@ namespace Facebook.Yoga
             if (baselineFunction != null)
             {
 #if (UNITY_IOS && !UNITY_EDITOR) || __IOS__
+                _managedBaseline = BaselineInternalIOS;
                 _ygNode.SetContext(this);
 #else
                 _managedBaseline = BaselineInternal;
 #endif
             }
+            else
+            {
+                _managedBaseline = null;
+            }
             Native.YGNodeSetBaselineFunc(_ygNode, _managedBaseline);
         }