From 4623dd72832f0252bcee95832cc51ce665cfee31 Mon Sep 17 00:00:00 2001
From: Jeroen Zwartepoorte <jeroen.zwartepoorte@iddinkgroup.com>
Date: Thu, 15 Dec 2022 15:52:32 +0100
Subject: [PATCH] Make the viewport addon compatible with composition.

---
 code/addons/viewport/src/Tool.tsx | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/code/addons/viewport/src/Tool.tsx b/code/addons/viewport/src/Tool.tsx
index 10c2a4dfadd0..ec14c0740fa3 100644
--- a/code/addons/viewport/src/Tool.tsx
+++ b/code/addons/viewport/src/Tool.tsx
@@ -58,7 +58,6 @@ const toLinks = memoize(50)((list: ViewportItem[], active: LinkBase, set, state,
     .filter(Boolean);
 });
 
-const iframeId = 'storybook-preview-iframe';
 const wrapperId = 'storybook-preview-wrapper';
 
 interface LinkBase {
@@ -208,7 +207,7 @@ export const ViewportTool: FC = memo(
           <ActiveViewportSize>
             <Global
               styles={{
-                [`#${iframeId}`]: {
+                [`iframe[data-is-storybook="true"]`]: {
                   margin: `auto`,
                   transition: 'width .3s, height .3s',
                   position: 'relative',