Fix suspense replaying forwardRefs (facebook#26385)

hansottowirtz · Apr 2, 2023 · 9829d02 · 9829d02
1 parent 0ae3480
commit 9829d02
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/packages/react-reconciler/src/ReactFiberWorkLoop.js b/packages/react-reconciler/src/ReactFiberWorkLoop.js
@@ -2386,7 +2386,10 @@ 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