Fix for custom controls

1 files changed, 7 insertions, 4 deletions

diff --git a/src/js/utils.js b/src/js/utils.js
index 38e3d402..c53293f4 100644
--- a/src/js/utils.js
+++ b/src/js/utils.js
@@ -306,12 +306,15 @@ const utils = {
     // Remove an element
     removeElement(element) {
         if (!utils.is.element(element) || !utils.is.element(element.parentNode)) {
-            return null;
+            return;
         }
 
-        element.parentNode.removeChild(element);
+        if (utils.is.nodeList(element) || utils.is.array(element)) {
+            Array.from(element).forEach(utils.removeElement);
+            return;
+        }
 
-        return element;
+        element.parentNode.removeChild(element);
     },
 
     // Remove all child elements
@@ -569,7 +572,7 @@ const utils = {
         }
 
         // If a nodelist is passed, call itself on each node
-        if (utils.is.nodeList(elements)) {
+        if (utils.is.nodeList(elements) || utils.is.array(elements)) {
             // Create listener for each node
             Array.from(elements).forEach(element => {
                 if (element instanceof Node) {