Can't assign roles on subuser creation

The check boxes to assign roles on subuser creation have disappeared in the latest dev release. Comparing the alpha and dev code it appears some of the code from the subuser_form_user_register_form_alter has been removed.

Current Dev code

function subuser_form_user_register_form_alter(&$form, &$form_state) {
  global $user;

  $form['subuser_relation'] = array(
    '#type' => 'checkbox',
    '#title' => t('User relation'),
    '#description' => t('Store a relationship that will allow for further management of the user.'),
    '#default_value' => $user->uid && variable_get('subuser_relation', TRUE),
    '#access' => $user->uid && user_access('override subuser relation'),
  );

  $form['#submit'][] = 'subuser_user_register_form_submit';
}

Alpha Code

  function subuser_form_user_register_form_alter(&$form, &$form_state) {
    global $user;

    // If the user does not have access to the roles field then filter the roles
    // field based on subuser permissions and display if more then one left. If
    // the user has 'administer permissions' then the #access will be set to TRUE
    // and they will have access to all roles, otherwise if the user has access
    // to this page through subuser then only provide them with the roles they
    // are allowed based on subuser. All users will have at least one role, but
    // that role may be 'authenticated user' which is not included in #options.
    if (!$form['account']['roles']['#access']) {
      foreach ($form['account']['roles']['#options'] as $rid => $role) {
        if (!user_access('create subuser ' . $rid)) {
          unset($form['account']['roles']['#options'][$rid]);
        }
      }
      $form['account']['roles']['#access'] = count($form['account']['roles']['#options']) > 0;
    }

    $form['subuser_relation'] = array(
      '#type' => 'checkbox',
      '#title' => t('User relation'),
      '#description' => t('Store a relationship that will allow for further management of the user.'),
      '#default_value' => $user->uid && variable_get('subuser_relation', TRUE),
      '#access' => $user->uid && user_access('override subuser relation'),
    );

    $form['#submit'][] = 'subuser_user_register_form_submit';
  }

Putting this bit back into the dev code appears to make everything work correctly but I'm not sure if this was removed on purpose or if there has been any changes that would require this be updated.

    // If the user does not have access to the roles field then filter the roles
    // field based on subuser permissions and display if more then one left. If
    // the user has 'administer permissions' then the #access will be set to TRUE
    // and they will have access to all roles, otherwise if the user has access
    // to this page through subuser then only provide them with the roles they
    // are allowed based on subuser. All users will have at least one role, but
    // that role may be 'authenticated user' which is not included in #options.
    if (!$form['account']['roles']['#access']) {
      foreach ($form['account']['roles']['#options'] as $rid => $role) {
        if (!user_access('create subuser ' . $rid)) {
          unset($form['account']['roles']['#options'][$rid]);
        }
      }
      $form['account']['roles']['#access'] = count($form['account']['roles']['#options']) > 0;
    }