Checkboxes and radios disappear in tableselect

Hi! At days I found one bug in this theme. I created custom form with 'tableselect' element and instead of checkboxes in the first column of table I found only empty div's in cells. I dig deeper into this problem and found the following lines in 'includes/form.inc'

function radix_preprocess_form_element(&$variables) {
  // Render the form element inside the label for checkbox and radio.
  if (in_array($variables['element']['#type'], ['checkbox', 'radio'])) {
    $variables['label']['#children'] = $variables['children'];
    unset($variables['children']);
  }
}

This code moves checkbox and radios into the labels. It is OK, but in tableselect we have checkboxes and radios without labels. So if label is invisible, it's children are not rendered too. So I propose the following solution:

if (in_array($variables['element']['#type'], ['checkbox', 'radio']) && $variables['label']['#title_display'] !== 'invisible') {

And here is a patch with this fix.