diff --git a/src/components/main/TreeModal.js b/src/components/main/TreeModal.js
index b3e15c268..39a80aec1 100644
--- a/src/components/main/TreeModal.js
+++ b/src/components/main/TreeModal.js
@@ -130,7 +130,7 @@ TreeModal.propTypes = {
   onClose: PropTypes.func.isRequired,
   title: PropTypes.string.isRequired,
   prevent: PropTypes.oneOf(Object.values(TREE_PREVENT_SELECTION)),
-  itemId: PropTypes.string,
+  itemId: PropTypes.oneOfType([ PropTypes.string, PropTypes.array ]),
   open: PropTypes.bool,
 };