diff --git a/editor_note.module b/editor_note.module
index 4ceaa75..2c0ec3a 100644
--- a/editor_note.module
+++ b/editor_note.module
@@ -101,6 +101,7 @@ function editor_note_menu() {
     'access arguments' => array(3, 4),
     'type' => MENU_NORMAL_ITEM,
     'file' => 'editor_note.pages.inc',
+    'theme callback' => 'ajax_base_page_theme',
   );
 
   $items['editor_note/remove/%ctools_js/%/%'] = array(
@@ -111,6 +112,7 @@ function editor_note_menu() {
     'access arguments' => array(3, 4),
     'type' => MENU_NORMAL_ITEM,
     'file' => 'editor_note.pages.inc',
+    'theme callback' => 'ajax_base_page_theme',
   );
 
   return $items;
diff --git a/editor_note.pages.inc b/editor_note.pages.inc
index be8eec6..e8a7899 100644
--- a/editor_note.pages.inc
+++ b/editor_note.pages.inc
@@ -29,9 +29,6 @@ function editor_note_confirm_edit_page($ajax = NULL, $note_id = NULL) {
       $output = $form_state['ajax_commands'];
     }
 
-    // Prevents adding system default theme CSS to the admin theme.
-    $_POST['ajax_page_state']['css'] = array();
-
     // Returns the ajax instructions to the browser via ajax_render().
     print ajax_render($output);
     drupal_exit();
@@ -220,9 +217,6 @@ function editor_note_confirm_remove_page($ajax = NULL, $note_id = NULL) {
       $output = $form_state['ajax_commands'];
     }
 
-    // Prevents adding system default theme CSS to the admin theme.
-    $_POST['ajax_page_state']['css'] = array();
-
     // Returns the ajax instructions to the browser via ajax_render().
     print ajax_render($output);
     drupal_exit();