fix review

RocketChat · Feb 14, 2023 · b963d94 · b963d94
1 parent 0331379
commit b963d94
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/ee/packages/ui-theming/src/hooks/useCreateStyleContainer.ts b/ee/packages/ui-theming/src/hooks/useCreateStyleContainer.ts
@@ -1,7 +1,11 @@
+import { useMemo } from 'react';
+
 export const useCreateStyleContainer = (id: string) => {
-	const refElement = document.getElementById('css-theme') || document.head.lastChild;
-	const styleElement = document.createElement('style');
-	styleElement.setAttribute('id', id);
-	document.head.insertBefore(styleElement, refElement);
-	return document.getElementById(id) || document.head.appendChild(document.createElement('style'));
+	return useMemo(() => {
+		const refElement = document.getElementById('css-theme') || document.head.lastChild;
+		const styleElement = document.createElement('style');
+		styleElement.setAttribute('id', id);
+		document.head.insertBefore(styleElement, refElement);
+		return document.getElementById(id) || document.head.appendChild(document.createElement('style'));
+	}, [id]);
 };