diff --git a/embeddable.module b/embeddable.module
index 6bc7459..e817296 100644
--- a/embeddable.module
+++ b/embeddable.module
@@ -271,6 +271,7 @@ function embeddable_pre_render_text_format($element) {
       "embeddable_embeds"=> array_keys($embeds)
     ), 'setting');
   //TODO: This needs to be moved to getting loaded via ajax to work in boxes / panes ect.
-  $page['content']['embeddable'] = drupal_get_form('embeddable_embed_form');
+  $element['embeddable'] = drupal_get_form('embeddable_embed_form');
+
   return $element;
 }