diff --git a/src/AnnotationCreation.js b/src/AnnotationCreation.js
index c1ab3dcaf5bfae400871889a3f44dcf7f27b5db1..049f8877d5068c3914c5fea8a88085a0d92eb421 100644
--- a/src/AnnotationCreation.js
+++ b/src/AnnotationCreation.js
@@ -7,7 +7,7 @@ import {
 import { Alarm, LastPage } from '@mui/icons-material';
 import Typography from '@mui/material/Typography';
 import ToggleButton from '@mui/lab/ToggleButton';
-import ToggleButtonGroup from '@mui/lab/ToggleButtonGroup';
+import ToggleButtonGroup from '@mui/material/ToggleButtonGroup';
 import RectangleIcon from '@mui/icons-material/CheckBoxOutlineBlank';
 import CircleIcon from '@mui/icons-material/RadioButtonUnchecked';
 import PolygonIcon from '@mui/icons-material/Timeline';
diff --git a/src/TextEditor.js b/src/TextEditor.js
index e157d2bd9c458b6faa6c3e27098a1fdc31ef5aa1..116de91be9a8d953a97d2f2933caf89efa7c339b 100644
--- a/src/TextEditor.js
+++ b/src/TextEditor.js
@@ -2,7 +2,7 @@ import React, { useState, useRef, useEffect } from 'react';
 import PropTypes from 'prop-types';
 import { Editor, EditorState, RichUtils } from 'draft-js';
 import ToggleButton from '@mui/lab/ToggleButton';
-import ToggleButtonGroup from '@mui/lab/ToggleButtonGroup';
+import ToggleButtonGroup from '@mui/material/ToggleButtonGroup';
 import BoldIcon from '@mui/icons-material/FormatBold';
 import ItalicIcon from '@mui/icons-material/FormatItalic';
 import { stateToHTML } from 'draft-js-export-html';