diff --git a/freeze-ui.js b/freeze-ui.js
index 3365ea8..9be509f 100644
--- a/freeze-ui.js
+++ b/freeze-ui.js
@@ -29,8 +29,8 @@
      * Unfreezes the UI.
      * No options here.
      */
-    window.UnFreezeUI = () => {
-        let element = document.querySelector('.freeze-ui');
+    window.UnFreezeUI = (selector) => {
+        let element = document.querySelector((selector?selector+' ':'')+'.freeze-ui');
         if (element) {
             element.classList.add('is-unfreezing');
             setTimeout(() => {
@@ -41,4 +41,4 @@
             }, 250);
         }
     };
-})();
\ No newline at end of file
+})();