diff --git a/domain_theme/domain_theme.module b/domain_theme/domain_theme.module
index 9aa8128..e9353b6 100644
--- a/domain_theme/domain_theme.module
+++ b/domain_theme/domain_theme.module
@@ -16,28 +16,19 @@
  */
 
 /**
- * Implements hook_custom_theme().
+ * Implements hook_domain_bootstrap_full().
  */
-function domain_theme_custom_theme() {
+function domain_theme_domain_bootstrap_full() {
   global $conf;
   $_domain = domain_get_domain();
 
   // Assign the theme selected, based on the active domain.
   $theme = domain_theme_lookup($_domain['domain_id']);
 
-  // Check for a custom admin theme.
-  if (isset($conf['admin_theme']) && path_is_admin(current_path()) && user_access('view the administration theme')) {
-    $theme = domain_theme_lookup($_domain['domain_id'], $conf['admin_theme']);
-    // No custom settings, but use the admin theme.
-    if ($theme == -1) {
-      return $conf['admin_theme'];
-    }
-  }
-
   // The above returns -1 on failure.
   if ($theme != -1) {
     domain_theme_set_variables($theme);
-    return $theme['theme'];
+    $conf['theme_default'] = $theme['theme'];
   }
 }