diff --git a/menu_trail_by_path.module b/menu_trail_by_path.module
index 8e4ab0f..99e3595 100644
--- a/menu_trail_by_path.module
+++ b/menu_trail_by_path.module
@@ -14,13 +14,13 @@
  */
 
 /**
- * Implements hook_page_delivery_callback_alter().
+ * Implements hook_init to ensure active-trail gets set.
  * 
  * This is the only hook that occurs after the page callback, but before
  * hook_page_build (when blocks are added). We're using this hook for its
  * timing, not its data.
  */
-function menu_trail_by_path_page_delivery_callback_alter() {
+function menu_trail_by_path_init() {
   global $language;
   $parent_candidates = _menu_trail_by_path_get_parent_candidates(drupal_get_path_alias());
   // Don't even bother if current page is root