diff --git a/packages/react-reconciler/src/__tests__/ReactLazy-test.internal.js b/packages/react-reconciler/src/__tests__/ReactLazy-test.internal.js
index 6efd405aba..bc0fdf1293 100644
--- a/packages/react-reconciler/src/__tests__/ReactLazy-test.internal.js
+++ b/packages/react-reconciler/src/__tests__/ReactLazy-test.internal.js
@@ -619,12 +619,12 @@ describe('ReactLazy', () => {
     expect(root).toMatchRenderedOutput('22');
 
     // Update
-    root.update(
-      <Suspense fallback={<Text text="Loading..." />}>
-        <LazyAdd inner={false} outer={false} />
-      </Suspense>,
-    );
     expect(() => {
+      root.update(
+        <Suspense fallback={<Text text="Loading..." />}>
+          <LazyAdd inner={false} outer={false} />
+        </Suspense>,
+      );
       root.unstable_flushAll();
     }).toWarnDev(
       'Invalid prop `inner` of type `boolean` supplied to `Add`, expected `number`.',
@@ -804,12 +804,12 @@ describe('ReactLazy', () => {
     expect(root).toMatchRenderedOutput('Inner default text');
 
     // Update
-    root.update(
-      <Suspense fallback={<Text text="Loading..." />}>
-        <LazyText text={null} />
-      </Suspense>,
-    );
     expect(() => {
+      root.update(
+        <Suspense fallback={<Text text="Loading..." />}>
+          <LazyText text={null} />
+        </Suspense>,
+      );
       root.unstable_flushAll();
     }).toWarnDev(
       'The prop `text` is marked as required in `T`, but its value is `null`',
diff --git a/packages/react-reconciler/src/__tests__/ReactMemo-test.internal.js b/packages/react-reconciler/src/__tests__/ReactMemo-test.internal.js
index dbbe0683d0..046dd395a4 100644
--- a/packages/react-reconciler/src/__tests__/ReactMemo-test.internal.js
+++ b/packages/react-reconciler/src/__tests__/ReactMemo-test.internal.js
@@ -332,16 +332,16 @@ describe('memo', () => {
         const Fn = React.memo(FnInner);
 
         // Mount
-        ReactNoop.render(<Fn inner="2" />);
         expect(() => {
+          ReactNoop.render(<Fn inner="2" />);
           ReactNoop.flush();
         }).toWarnDev(
           'Invalid prop `inner` of type `string` supplied to `FnInner`, expected `number`.',
         );
 
         // Update
-        ReactNoop.render(<Fn inner={false} />);
         expect(() => {
+          ReactNoop.render(<Fn inner={false} />);
           ReactNoop.flush();
         }).toWarnDev(
           'Invalid prop `inner` of type `boolean` supplied to `FnInner`, expected `number`.',
@@ -358,20 +358,20 @@ describe('memo', () => {
         // Mount
         expect(() => {
           ReactNoop.render(<Fn outer="3" />);
+          ReactNoop.flush();
         }).toWarnDev(
           // Outer props are checked in createElement
           'Invalid prop `outer` of type `string` supplied to `FnInner`, expected `number`.',
         );
-        ReactNoop.flush();
 
         // Update
         expect(() => {
           ReactNoop.render(<Fn outer={false} />);
+          ReactNoop.flush();
         }).toWarnDev(
           // Outer props are checked in createElement
           'Invalid prop `outer` of type `boolean` supplied to `FnInner`, expected `number`.',
         );
-        ReactNoop.flush();
       });
 
       it('validates nested propTypes declarations', () => {
@@ -394,14 +394,9 @@ describe('memo', () => {
         // Mount
         expect(() => {
           ReactNoop.render(<Outer inner="2" middle="3" outer="4" />);
-        }).toWarnDev(
-          // Outer props are checked in createElement
-          'Invalid prop `outer` of type `string` supplied to `Inner`, expected `number`.',
-        );
-        expect(() => {
           ReactNoop.flush();
         }).toWarnDev([
-          // There are no actual elements created for these, so they're checked in reconciler.
+          'Invalid prop `outer` of type `string` supplied to `Inner`, expected `number`.',
           'Invalid prop `middle` of type `string` supplied to `Inner`, expected `number`.',
           'Invalid prop `inner` of type `string` supplied to `Inner`, expected `number`.',
         ]);
@@ -411,14 +406,9 @@ describe('memo', () => {
           ReactNoop.render(
             <Outer inner={false} middle={false} outer={false} />,
           );
-        }).toWarnDev(
-          // Outer props are checked in createElement
-          'Invalid prop `outer` of type `boolean` supplied to `Inner`, expected `number`.',
-        );
-        expect(() => {
           ReactNoop.flush();
         }).toWarnDev([
-          // There are no actual elements created for these, so they're checked in reconciler.
+          'Invalid prop `outer` of type `boolean` supplied to `Inner`, expected `number`.',
           'Invalid prop `middle` of type `boolean` supplied to `Inner`, expected `number`.',
           'Invalid prop `inner` of type `boolean` supplied to `Inner`, expected `number`.',
         ]);