diff --git a/src/Plugin/views/field/WeightSelector.php b/src/Plugin/views/field/WeightSelector.php
index ae9652d..0188dc5 100644
--- a/src/Plugin/views/field/WeightSelector.php
+++ b/src/Plugin/views/field/WeightSelector.php
@@ -91,8 +91,8 @@ class WeightSelector extends FieldPluginBase {
   }
 
   function viewsFormSubmit(array &$form, FormStateInterface $form_state) {
-    $rows = $form_state->getValue('field_weight');
     $field_name = $form_state->getValue('views_field');
+    $rows = $form_state->getValue($field_name);
 
     foreach ($rows as $row) {
       $entity = $row['entity'];