diff --git a/src/widgets/button/OpenQuizButton.tsx b/src/widgets/button/OpenQuizButton.tsx
index 1567a0c..8ad5ad8 100644
--- a/src/widgets/button/OpenQuizButton.tsx
+++ b/src/widgets/button/OpenQuizButton.tsx
@@ -14,7 +14,7 @@ interface Props {
 export default function OpenQuizButton({ selector, quizId, fixedSide }: Props) {
     const [isQuizDialogOpen, setIsQuizDialogOpen] = useState<boolean>(false);
 
-    const portalContainer = !fixedSide && selector ? document.getElementById(selector)! : document.body;
+    const portalContainer = !fixedSide && selector ? document.querySelector(selector)! : document.body;
 
     return createPortal(
         <ThemeProvider theme={lightTheme}>
diff --git a/src/widgets/container/ContainerWidget.tsx b/src/widgets/container/ContainerWidget.tsx
index 63b4d62..06a87cd 100644
--- a/src/widgets/container/ContainerWidget.tsx
+++ b/src/widgets/container/ContainerWidget.tsx
@@ -9,7 +9,7 @@ export class ContainerWidget {
         quizId: string;
         selector: string;
     }) {
-        const element = document.getElementById(selector);
+        const element = document.querySelector(selector);
         if (!element) throw new Error("Element for widget doesn't exist");
 
         this.root = createRoot(element);