diff --git a/src/components/modals/AddModal.jsx b/src/components/modals/AddModal.jsx
index 0cdac51..abf1fb7 100644
--- a/src/components/modals/AddModal.jsx
+++ b/src/components/modals/AddModal.jsx
@@ -53,7 +53,7 @@ class AddModal extends React.Component {
     }
   }
 
-  UNSAFE_componentWillUpdate(nextProps, nextState) {
+  shouldComponentUpdate(nextProps, nextState) {
     // Check if source is valid for new type
     const oldType = this.state.type;
     const newType = nextState.type;
@@ -75,6 +75,7 @@ class AddModal extends React.Component {
         source: ""
       });
     }
+    return true;
   }
 
   getLayersForSource(source) {