Fix suspense replaying forwardRefs (facebook#26385)

hansottowirtz · Mar 17, 2023 · 3864bae · 3864bae
1 parent 5530855
commit 3864bae
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/packages/react-reconciler/src/ReactFiberWorkLoop.js b/packages/react-reconciler/src/ReactFiberWorkLoop.js
@@ -2375,7 +2375,7 @@ function replaySuspendedUnitOfWork(unitOfWork: Fiber): void {
       // TODO: Consider moving this switch statement into that module. Also,
       // could maybe use this as an opportunity to say `use` doesn't work with
       // `defaultProps` :)
-      const Component = unitOfWork.type;
+      const Component = unitOfWork.tag === FunctionComponent ? unitOfWork.type : unitOfWork.type.render;
       const unresolvedProps = unitOfWork.pendingProps;
       const resolvedProps =
         unitOfWork.elementType === Component