diff --git a/src/MathInput.js b/src/MathInput.js
index 236eb6a4..0a06967d 100644
--- a/src/MathInput.js
+++ b/src/MathInput.js
@@ -36,15 +36,12 @@ class MatrixSizePicker extends React.Component {
}
// https://blog.logrocket.com/controlling-tooltips-pop-up-menus-using-compound-components-in-react-ccedc15c7526/
- componentDidUpdate() {
- setTimeout(() => {
- if(this.state.showMenu){
- window.addEventListener('click', this.closeMenu);
- }
- else{
- window.removeEventListener('click', this.closeMenu);
- }
- }, 0);
+ componentDidMount() {
+ window.addEventListener('click', this.closeMenu);
+ }
+
+ componentWillUnmount() {
+ window.removeEventListener('click', this.closeMenu);
}
render() {
@@ -113,12 +110,14 @@ class MatrixSizePicker extends React.Component {