diff --git a/apple/LayoutReanimation/REASwizzledUIManager.mm b/apple/LayoutReanimation/REASwizzledUIManager.mm
index 3196898536d3..98e765bdacd4 100644
--- a/apple/LayoutReanimation/REASwizzledUIManager.mm
+++ b/apple/LayoutReanimation/REASwizzledUIManager.mm
@@ -220,8 +220,8 @@ - (RCTViewManagerUIBlock)reanimated_uiBlockWithLayoutUpdateForRootView:(RCTRootS
       CGSize contentSize = shadowView.layoutMetrics.frame.size;
 
       RCTExecuteOnMainQueue(^{
-        NSMutableDictionary<NSNumber *, REAUIView *> *viewRegistry = [self valueForKey:@"_viewRegistry"];
-        REAUIView *view = viewRegistry[reactTag];
+        RCTUIManager *uiManager = [self valueForKey:@"_uiManager"];
+        REAUIView *view = [uiManager viewForReactTag:(NSNumber *)reactTag];
         RCTAssert(view != nil, @"view (for ID %@) not found", reactTag);
 
         RCTRootView *rootView = (RCTRootView *)[view superview];
@@ -245,7 +245,7 @@ - (RCTViewManagerUIBlock)reanimated_uiBlockWithLayoutUpdateForRootView:(RCTRootS
     for (NSNumber *reactTag in reactTags) {
       RCTFrameData frameData = frameDataArray[index++];
 
-      REAUIView *view = viewRegistry[reactTag];
+      REAUIView *view = [uiManager viewForReactTag:(NSNumber *)reactTag];
       CGRect frame = frameData.frame;
 
       UIUserInterfaceLayoutDirection layoutDirection = frameData.layoutDirection;
@@ -331,7 +331,7 @@ - (RCTViewManagerUIBlock)reanimated_uiBlockWithLayoutUpdateForRootView:(RCTRootS
     index = 0;
     for (NSNumber *reactTag in reactTags) {
       RCTFrameData frameData = frameDataArray[index++];
-      REAUIView *view = viewRegistry[reactTag];
+      REAUIView *view = [uiManager viewForReactTag:(NSNumber *)reactTag];
       BOOL isNew = frameData.isNew;
       CGRect frame = frameData.frame;
 
diff --git a/apple/REANodesManager.mm b/apple/REANodesManager.mm
index 2b9844049fa8..4fa8f448a4f5 100644
--- a/apple/REANodesManager.mm
+++ b/apple/REANodesManager.mm
@@ -410,7 +410,7 @@ - (void)configureUiProps:(nonnull NSSet<NSString *> *)uiPropsSet
 
 - (BOOL)isNativeViewMounted:(NSNumber *)viewTag
 {
-  REAUIView *view = _viewRegistry[viewTag];
+  REAUIView *view = [_uiManager viewForReactTag:(NSNumber *)viewTag];
   if (view.superview != nil) {
     return YES;
   }