From 3398831571ee98e4132e0dc4ec53818051ff1133 Mon Sep 17 00:00:00 2001
From: Jerko Steiner <jerko.steiner@gmail.com>
Date: Mon, 16 Sep 2019 23:07:07 +0700
Subject: [PATCH] Fix TestContainer not being used

---
 packages/client/src/test-utils/TestUtils.tsx | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/packages/client/src/test-utils/TestUtils.tsx b/packages/client/src/test-utils/TestUtils.tsx
index 2317682..f3846d4 100644
--- a/packages/client/src/test-utils/TestUtils.tsx
+++ b/packages/client/src/test-utils/TestUtils.tsx
@@ -32,10 +32,9 @@ export class TestUtils {
 
   render(jsx: JSX.Element) {
     const $div = document.createElement('div')
-    ReactDOM.render(<TestContainer>{jsx}</TestContainer>, $div)
     const component = ReactDOM.render(
-      <div>{jsx}</div>, $div) as unknown as TestContainer
-    const node = component.ref.current!
+      <TestContainer>{jsx}</TestContainer>, $div) as unknown as TestContainer
+    const node = component.ref.current!.children[0]
     return {
       component,
       node,