diff --git a/js/filter_checkboxes.js b/js/filter_checkboxes.js
index f68bbfc..3519f17 100644
--- a/js/filter_checkboxes.js
+++ b/js/filter_checkboxes.js
@@ -3,7 +3,7 @@
     attach: function (context, settings) {
       var $checkboxesElement = $('.checkboxes-filter-element', context);
       function filterCheckboxes (filterText, $element) {
-        $('div.form-item.form-type-checkbox', $element).each(function() {
+        $('div.form-item.form-type-checkbox, div.js-form-item.js-form-type-checkbox', $element).each(function() {
           if (filterText === '') {
             $(this).show();
           } else {
@@ -19,7 +19,7 @@
       $checkboxesElement.once('checkboxes-filter').each(function () {
         let currentElement = $(this);
         currentElement.prepend(
-          $('<input type="text" class="form-text" placeholder="' + Drupal.t('Filter') + '">').on('blur', function () {
+          $('<input type="text" class="form-text filter-input" placeholder="' + Drupal.t('Filter') + '">').on('blur', function () {
             filterCheckboxes($(this).val(), currentElement);
           }).on('keyup', function () {
             filterCheckboxes($(this).val(), currentElement);