diff --git a/java/jni/YGJNIVanilla.cpp b/java/jni/YGJNIVanilla.cpp
index 698ab561..65343821 100644
--- a/java/jni/YGJNIVanilla.cpp
+++ b/java/jni/YGJNIVanilla.cpp
@@ -196,7 +196,7 @@ static void jni_YGConfigSetLoggerJNI(
       delete context;
       YGConfigSetContext(config, nullptr);
     }
-    config->setLogger(nullptr);
+    YGConfigSetLogger(config, nullptr);
   }
 }
 
diff --git a/java/tests/com/facebook/yoga/YogaLoggerTest.java b/java/tests/com/facebook/yoga/YogaLoggerTest.java
index 17316f6e..f88409d6 100644
--- a/java/tests/com/facebook/yoga/YogaLoggerTest.java
+++ b/java/tests/com/facebook/yoga/YogaLoggerTest.java
@@ -9,10 +9,62 @@ package com.facebook.yoga;
 
 import org.junit.Test;
 import java.lang.ref.WeakReference;
+import java.util.List;
+import java.util.ArrayList;
 
+import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.fail;
 
 public class YogaLoggerTest {
+
+  @Test
+  public void testRemovingLoggerFromConfig() throws Exception {
+    final List<String> logs = new ArrayList<>();
+
+    final YogaConfig config = YogaConfigFactory.create();
+    YogaLogger logger = new YogaLogger() {
+      @Override
+      public void log(YogaLogLevel level, String message) {
+        logs.add(message);
+      }
+    };
+    config.setLogger(logger);
+
+    final YogaNode root = YogaNodeFactory.create(config);
+    root.setFlexDirection(YogaFlexDirection.ROW);
+    root.setAlignItems(YogaAlign.BASELINE);
+
+    final YogaNode child1 = YogaNodeFactory.create(config);
+    root.addChildAt(child1, 0);
+
+    final YogaNode child2 = YogaNodeFactory.create(config);
+    child2.setBaselineFunction(new YogaBaselineFunction() {
+        public float baseline(YogaNode node, float width, float height) {
+          return Float.NaN;
+        }
+    });
+    root.addChildAt(child2, 1);
+
+    assertEquals(logs.size(), 0);
+    try {
+      root.calculateLayout(YogaConstants.UNDEFINED, YogaConstants.UNDEFINED);
+      fail("Expected calculateLayout to throw");
+    } catch (IllegalStateException e) {
+    }
+
+    assertEquals(logs.size(), 1);
+
+    config.setLogger(null);
+
+    try {
+      root.calculateLayout(YogaConstants.UNDEFINED, YogaConstants.UNDEFINED);
+      fail("Expected calculateLayout to throw again");
+    } catch (IllegalStateException e) {
+    }
+
+    assertEquals(logs.size(), 1);
+  }
+
   @Test
   public void testLoggerLeak() throws Exception {
     final YogaConfig config = YogaConfigFactory.create();