Reordering Weights causes order to be lost

This code inside the multiple_fields_remove_button_submit_handler function causes the overall order of items to be lost and should be reworked.

(line 444 in multiple_fields_remove_button.module)

  if ($input && is_array($input)) {
    // Sort by weight.
    uasort($input, '_field_multiple_value_form_sort_helper');

    // Reweight everything in the correct order.
    $weight = -1 * $field_state['items_count'];
    foreach ($input as $key => $item) {
      if ($item) {
        $input[$key]['_weight'] = $weight++;
      }
    }
    NestedArray::setValue($formInputs, $address, $input);
    $form_state->setUserInput($formInputs);
  }

Removing it doesn't seem to cause a problem, the drag and drop functionality automatically reassigns assigns correct weights, and on saving the page the weights are also reordered to be correct without losing the user's original order.