minor cleanup

src/renderHookToSnapshotStream.ts → src/renderHookToSnapshotStream.tsx

@@ -3,7 +3,7 @@ import { createRenderStream } from "./renderStream/createRenderStream.js";
 import type { NextRenderOptions } from "./renderStream/createRenderStream.js";
 
 import { Render } from "./renderStream/Render.js";
-import { createElement } from "react";
+import { createElement } from "rehackt";
 import { Assertable, assertableSymbol, markAssertable } from "./assertable.js";
 
 export interface SnapshotStream<Snapshot, Props> extends Assertable {

src/renderStream/createRenderStream.tsx

@@ -221,15 +221,12 @@ export function createRenderStream<Snapshot extends ValidSnapshot = void>({
     return baseRender(ui, {
       ...options,
       wrapper: (props) => {
-        let elem: React.ReactNode = React.createElement(
-          Wrapper,
-          undefined,
-          props.children
+        const ParentWrapper = options?.wrapper || React.Fragment;
+        return (
+          <ParentWrapper>
+            <Wrapper>{props.children}</Wrapper>
+          </ParentWrapper>
         );
-        if (options?.wrapper) {
-          elem = React.createElement(options.wrapper, undefined, elem);
-        }
-        return elem;
       },
     });
   }) as typeof baseRender;
@@ -358,15 +355,15 @@ export function useTrackRenders({ name }: { name?: string } = {}) {
 
   if (!component) {
     throw new Error(
-      "useTrackRender: Unable to determine component. Please ensure the hook is called inside a rendered component or provide a `name` option."
+      "useTrackRenders: Unable to determine component. Please ensure the hook is called inside a rendered component or provide a `name` option."
     );
   }
 
   const ctx = useRenderStreamContext();
 
   if (!ctx) {
     throw new Error(
-      "useTrackComponentRender: A Render Stream must be created and rendered to track component renders"
+      "useTrackRenders: A Render Stream must be created and rendered to track component renders"
     );
   }