diff --git a/tests/utils.tsx b/tests/utils.tsx
index 8a567d6..b13bf00 100644
--- a/tests/utils.tsx
+++ b/tests/utils.tsx
@@ -9,7 +9,9 @@ export function createTrackingWrapper(defaultProps: TrackingProps = {}) {
 
     return (
       <Tracking {...props}>
-        <TrackingHandler onHandle={spy}>{children}</TrackingHandler>
+        {ctx =>
+          <TrackingHandler onHandle={spy}>{typeof children === 'function' ? children(ctx) : children}</TrackingHandler>
+        }
       </Tracking>
     );
   };