diff --git a/java/jni/YGJNI.cpp b/java/jni/YGJNI.cpp
index a7fef2e6..18de9246 100644
--- a/java/jni/YGJNI.cpp
+++ b/java/jni/YGJNI.cpp
@@ -7,6 +7,7 @@
 #include <fb/fbjni.h>
 #include <yoga/YGNode.h>
 #include <yoga/Yoga.h>
+#include <yoga/Yoga-internal.h>
 #include <yoga/log.h>
 #include <cstdint>
 #include <cstring>
diff --git a/yoga/Yoga-internal.h b/yoga/Yoga-internal.h
index f5cd6136..be815921 100644
--- a/yoga/Yoga-internal.h
+++ b/yoga/Yoga-internal.h
@@ -29,6 +29,8 @@ void YGNodeCalculateLayoutWithContext(
     YGDirection ownerDirection,
     void* layoutContext);
 
+void YGSetUsedCachedEntries(size_t);
+
 YG_EXTERN_C_END
 
 namespace facebook {
diff --git a/yoga/Yoga.h b/yoga/Yoga.h
index 7132a934..753a2f9f 100644
--- a/yoga/Yoga.h
+++ b/yoga/Yoga.h
@@ -411,8 +411,6 @@ WIN_EXPORT float YGRoundValueToPixelGrid(
     const bool forceCeil,
     const bool forceFloor);
 
-void YGSetUsedCachedEntries(size_t);
-
 YG_EXTERN_C_END
 
 #ifdef __cplusplus