diff --git a/edit_section.module b/edit_section.module
index cfc2c9f..8c59689 100644
--- a/edit_section.module
+++ b/edit_section.module
@@ -157,6 +157,17 @@ function edit_section_form($form, $form_state, $node, $section_id) {
     '#type' => 'text_format',
     '#default_value' => substr($body[0]['value'], $offsets['start'], $offsets['end'] - $offsets['start'])
   );
+
+  if (module_exists('better_formats')) {
+    $form['section'] += array
+    (
+      '#entity_type' => 'node',
+      '#entity' => $node,
+      '#bundle' => $node->type,
+      '#field_name' => 'body',
+      '#format' => $body[0]['format'],
+    );
+  }
   
   $node_options = variable_get('node_options_'. $node->type, array('status', 'promote'));
   $revision = in_array('revision', $node_options);