Index: modules/hs_menu.module
===================================================================
RCS file: /cvs/drupal-contrib/contributions/modules/hierarchical_select/modules/hs_menu.module,v
retrieving revision 1.3
diff -u -F^f -r1.3 hs_menu.module
--- modules/hs_menu.module	16 Jul 2008 21:12:23 -0000	1.3
+++ modules/hs_menu.module	26 Jul 2008 10:37:48 -0000
@@ -14,26 +14,16 @@
  * Implementation of hook_form_alter().
  */
 function hs_menu_form_alter($form_id, &$form) {
+  if (isset($form['type']) && $form['type']['#value'] .'_node_form' == $form_id) {
+    unset($form['menu']['pid']['#options']);
+    $form['menu']['pid']['#type'] = 'hierarchical_select';
+    _hs_menu_apply($form['menu']['pid'], NULL);
+  }
+  
   if ($form_id == 'menu_edit_item_form') {
     unset($form['pid']['#options']);
     $form['pid']['#type'] = 'hierarchical_select';
-    $form['pid']['#config'] = array(
-      'module' => 'hs_menu',
-      'params' => array(
-        'mid' => $form['mid']['#value'],
-      ),
-      'save_lineage'    => 0,
-      'enforce_deepest' => 0,
-      'level_labels' => array(
-        'status' => 0,
-      ),
-      'dropbox' => array(
-        'status' => 0,
-      ),
-      'editability' => array(
-        'status' => 0,
-      ),
-    );
+    _hs_menu_apply($form['pid'], NULL);
   }
 }
 
@@ -149,3 +139,27 @@ function hs_menu_hierarchical_select_imp
     'entity type'    => t('N/A'),
   );
 }
+
+
+//----------------------------------------------------------------------------
+// Private functions.
+
+function _hs_menu_apply(&$form, $mid) {
+  $form['#config'] = array(
+    'module' => 'hs_menu',
+    'params' => array(
+      'mid' => $mid,
+    ),
+    'save_lineage'    => 0,
+    'enforce_deepest' => 0,
+    'level_labels' => array(
+      'status' => 0,
+    ),
+    'dropbox' => array(
+      'status' => 0,
+    ),
+    'editability' => array(
+      'status' => 0,
+    ),
+  );
+}