diff --git a/apple/LayoutReanimation/REASwizzledUIManager.mm b/apple/LayoutReanimation/REASwizzledUIManager.mm
index 98e765bdacd4..16a25aea62b2 100644
--- a/apple/LayoutReanimation/REASwizzledUIManager.mm
+++ b/apple/LayoutReanimation/REASwizzledUIManager.mm
@@ -220,8 +220,7 @@ - (RCTViewManagerUIBlock)reanimated_uiBlockWithLayoutUpdateForRootView:(RCTRootS
       CGSize contentSize = shadowView.layoutMetrics.frame.size;
 
       RCTExecuteOnMainQueue(^{
-        RCTUIManager *uiManager = [self valueForKey:@"_uiManager"];
-        REAUIView *view = [uiManager viewForReactTag:(NSNumber *)reactTag];
+        REAUIView *view = [originalSelf viewForReactTag:(NSNumber *)reactTag];
         RCTAssert(view != nil, @"view (for ID %@) not found", reactTag);
 
         RCTRootView *rootView = (RCTRootView *)[view superview];
@@ -245,7 +244,7 @@ - (RCTViewManagerUIBlock)reanimated_uiBlockWithLayoutUpdateForRootView:(RCTRootS
     for (NSNumber *reactTag in reactTags) {
       RCTFrameData frameData = frameDataArray[index++];
 
-      REAUIView *view = [uiManager viewForReactTag:(NSNumber *)reactTag];
+      REAUIView *view = [originalSelf viewForReactTag:(NSNumber *)reactTag];
       CGRect frame = frameData.frame;
 
       UIUserInterfaceLayoutDirection layoutDirection = frameData.layoutDirection;
@@ -331,7 +330,7 @@ - (RCTViewManagerUIBlock)reanimated_uiBlockWithLayoutUpdateForRootView:(RCTRootS
     index = 0;
     for (NSNumber *reactTag in reactTags) {
       RCTFrameData frameData = frameDataArray[index++];
-      REAUIView *view = [uiManager viewForReactTag:(NSNumber *)reactTag];
+      REAUIView *view = [originalSelf viewForReactTag:(NSNumber *)reactTag];
       BOOL isNew = frameData.isNew;
       CGRect frame = frameData.frame;