search theme form broken

This should be a simple fix but I don't understand it.

It looks like the callback arguments don't match up. The theme function is not used at all and preventing me from converting search module into template files.

function search_box(&$form_state, $form_id) {
  // Use search_keys instead of keys to avoid ID conflicts with the search block.
  $form[$form_id .'_keys'] = array(
    '#type' => 'textfield',
    '#size' => 15,
    '#default_value' => '',
    '#attributes' => array('title' => t('Enter the terms you wish to search for.')),
  );
  $form['submit'] = array('#type' => 'submit', '#value' => t('Search'));
  // Always go to the search page since the search form is not guaranteed to be
  // on every page.
  $form['#action'] = url('search/node');
  $form['#submit'][] = 'search_box_form_submit';
  $form['#validate'][] = 'search_box_form_validate';
  $form['#theme'] = 'search_box_form';

  return $form;
}

function search_forms() {
  $forms['search_theme_form']= array(
    'callback' => 'search_box',
    'callback arguments' => array('search_theme_form'),
  );
  $forms['search_block_form']= array(
    'callback' => 'search_box',
    'callback arguments' => array('search_block_form'),
  );
  return $forms;
}

Can anyone spare a free moment and provide a patch? Or just tell me how to fix it and I'll roll it into the conversion. http://drupal.org/node/164032

thanks