diff --git a/client/src/components/Draftail/blocks/MediaBlock.test.js b/client/src/components/Draftail/blocks/MediaBlock.test.js
index 70609ae19..6db6621a8 100644
--- a/client/src/components/Draftail/blocks/MediaBlock.test.js
+++ b/client/src/components/Draftail/blocks/MediaBlock.test.js
@@ -83,8 +83,7 @@ describe('MediaBlock', () => {
 
       expect(
         wrapper
-          .find('Portal')
-          .instance().portal
+          .find('Portal > Portal').prop('containerInfo')
       ).toMatchSnapshot();
     });
 
@@ -110,9 +109,8 @@ describe('MediaBlock', () => {
 
       expect(
         wrapper
-          .find('Portal')
-          .instance()
-          .portal.querySelector('.Tooltip').className
+          .find('.Tooltip')
+          .prop('className')
       ).toBe('Tooltip Tooltip--left');
     });
 
diff --git a/client/src/components/Portal/__snapshots__/Portal.test.js.snap b/client/src/components/Portal/__snapshots__/Portal.test.js.snap
index 0e20a34e8..ed30d7824 100644
--- a/client/src/components/Portal/__snapshots__/Portal.test.js.snap
+++ b/client/src/components/Portal/__snapshots__/Portal.test.js.snap
@@ -14,7 +14,15 @@ exports[`Portal #children 1`] = `
   }
   onClose={[Function]}
 >
-  Test!
+  <Portal
+    containerInfo={
+      <div>
+        Test!
+      </div>
+    }
+  >
+    Test!
+  </Portal>
 </Portal>
 `;
 
@@ -31,5 +39,9 @@ exports[`Portal empty 1`] = `
     </body>
   }
   onClose={[Function]}
-/>
+>
+  <Portal
+    containerInfo={<div />}
+  />
+</Portal>
 `;