gistfile1.txt

diff --git a/js/update-slots.js b/js/update-slots.js
index 041bcdf..8f02c35 100644
--- a/js/update-slots.js
+++ b/js/update-slots.js
@@ -31,7 +31,7 @@
 
   Drupal.behaviors.vaa_update_slots = {
     attach: function (context, settings) {
-      $('.vaa-slots > div').once('vaa_init').each(function () {
+      $(once('vaa_init', '.vaa-slots > div', context)).each(function () {
         const $slot = $(this);
         const $parent = $slot.parent().parent().closest('fieldset');
 
diff --git a/viewfield_argument_helper.module b/viewfield_argument_helper.module
index 48a9355..8b6a5b1 100644
--- a/viewfield_argument_helper.module
+++ b/viewfield_argument_helper.module
@@ -27,7 +27,7 @@ function viewfield_argument_helper_help($route_name, RouteMatchInterface $route_
 /**
  * Implements hook_field_widget_viewfield_select_form_alter().
  */
-function viewfield_argument_helper_field_widget_viewfield_select_form_alter(&$element, FormStateInterface $form_state, $context) {
+function viewfield_argument_helper_field_widget_single_element_viewfield_select_form_alter(&$element, FormStateInterface $form_state, $context) {
   if (!\Drupal::currentUser()->hasPermission('use viewfield argument helper')) {
     return;
   }