Incorrect user assigned to new profiles when using address book route

Moving to commerce. It looks like we need to update the functional js tests to have a user create a profile for a different user than the one that's logged in.

The following code fixes the issue until we get this resolved.

/**
 * Implements hook_ENTITY_TYPE_prepare_form().
 */
function MY_MODULE_profile_prepare_form(\Drupal\Core\Entity\EntityInterface $entity, $operation, \Drupal\Core\Form\FormStateInterface $form_state) {
  $route = \Drupal::routeMatch();
  if (!empty($route) && $route->getRouteName() == 'commerce_order.address_book.add_form') {
    $user = $route->getParameter('user');
    if ($user->id() > 0) {
      $entity->set('uid', $user->id());
    }
  }
}