Avoid LivePreviewButton appearing a moment later

Automattic · Aug 2, 2023 · 058c7fe · 058c7fe
1 parent 8dad318
commit 058c7fe
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/client/my-sites/theme/main.jsx b/client/my-sites/theme/main.jsx
@@ -616,6 +616,8 @@ class ThemeSheet extends Component {
 			isLoggedIn,
 			themeId,
 			siteId,
+			isWpcomTheme,
+			isWporg,
 		} = this.props;
 		const placeholder = <span className="theme__sheet-placeholder">loading.....</span>;
 		const title = name || placeholder;
@@ -643,7 +645,14 @@ class ThemeSheet extends Component {
 							( this.shouldRenderUnlockStyleButton()
 								? this.renderUnlockStyleButton()
 								: this.renderButton() ) }
-						<LivePreviewButton themeId={ themeId } siteId={ siteId } />
+						<LivePreviewButton
+							themeId={ themeId }
+							/**
+							 * Pass the siteId that QueryCanonicalTheme component will use to fetch the theme.
+							 * This avoids LivePreviewButton appearing a moment later.
+							 */
+							siteId={ ( isWpcomTheme && 'wpcom' ) || ( isWporg && 'wporg' ) || siteId }
+						/>
 						{ this.shouldRenderPreviewButton() && (
 							<Button
 								onClick={ ( e ) => {