Missing contextual links when using "Menu selected by active page" option

Changed description to better describe the issue.

My apologies, I'm still a bit of a newb when it comes to these forums. Yes, I've patched this up for myself by changing hook_block_view() to read the data variable rather than the config. I didn't make a patch file for it but the change is fairly strait forward. Thanks!

function menu_block_block_view($delta = '') {
  $config = menu_block_get_config($delta);
  $data = menu_tree_build($config);
  // Add contextual links for this block.
  if (!empty($data['content'])) {
    if (in_array($data['content']['#config']['menu_name'], array_keys(menu_get_menus()))) {
      $data['content']['#contextual_links']['menu_block'] = array('admin/structure/menu/manage', array($data['content']['#config']['menu_name']));
    }
    elseif (strpos($data['content']['#config']['menu_name'], 'book-toc-') === 0) {
      $node = str_replace('book-toc-', '', $data['content']['#config']['menu_name']);
      $data['content']['#contextual_links']['menu_block'] = array('admin/content/book', array($node));
    }
  }
  return $data;
}

Thanks!