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