diff --git a/.changeset/three-kiwis-turn.md b/.changeset/three-kiwis-turn.md
index d5802fa586..eb67cbbbdc 100644
--- a/.changeset/three-kiwis-turn.md
+++ b/.changeset/three-kiwis-turn.md
@@ -12,8 +12,8 @@ import {
 
 const tester = createExtensionTester(extension);
 
-const { getByTestId } = renderInTestApp(tester.element());
+const { getByTestId } = renderInTestApp(tester.reactElement());
 
 // or if you're not using `coreExtensionData.reactElement` as the output ref
-const { getByTestId } = renderInTestApp(tester.data(myComponentRef));
+const { getByTestId } = renderInTestApp(tester.get(myComponentRef));
 ```