diff --git a/sites/all/modules/rdfx/rdfui/rdfui.module b/sites/all/modules/rdfx/rdfui/rdfui.module
index 7fa0328..7279395 100644
--- a/sites/all/modules/rdfx/rdfui/rdfui.module
+++ b/sites/all/modules/rdfx/rdfui/rdfui.module
@@ -341,7 +341,7 @@ function rdfui_form_field_ui_field_edit_form_submit($form, &$form_state) {
  */
 function rdfui_form_node_type_form_submit($form, &$form_state) {
   $entity_type = 'node';
-  $bundle = $form_state['input']['type'];
+  $bundle = $form_state['build_info']['args'][0]->type;
   // We only need to call _rdfui_mapping_save once because it checks whether
   // rdf_type is set in the form before saving the field value.
   _rdfui_mapping_save($form_state, $entity_type, $bundle, 'title');
