New Comment Tag on Answers incorrect due to history table not updated

I have an issue with the new comment tag always displays for answers when viewing the questions with associated answers page. I found that when one views a question and answers page the history table is updated only for the answers_question type, and not for the associated answers_answer type.

I added this code to answers module, answers_node_view()

global $user;
      if ($user->uid) {
        $result = db_select('field_data_answers_related_question', 'rq')
          ->fields('rq', array('entity_id'))
          ->condition('answers_related_question_target_id', $node->nid)
          ->execute()
          ->fetchAssoc();
        foreach($result as $row) {
          db_merge('history')
            ->key(array(
              'uid' => $user->uid,
              'nid' => $result,
            ))
            ->fields(array('timestamp' => REQUEST_TIME))
            ->execute();
        }
      }

It updates the history table for the answers_answer type correctly, yet the new comment marker now does not display at all. I suspect that I am on the right track, but that the history table should be updated later in the request.

Using hook_exit() seems to work:

function answers_exit() {
   $router_item  = menu_get_object();
  if (isset($router_item) && $router_item->type == "answers_question") {
    global $user;
    if ($user->uid) {
      $result = db_select('field_data_answers_related_question', 'rq')
        ->fields('rq', array('entity_id'))
        ->condition('answers_related_question_target_id', $router_item->nid)
        ->execute()
        ->fetchAssoc();
      foreach($result as $row) {
        db_merge('history')
          ->key(array(
            'uid' => $user->uid,
            'nid' => $result,
          ))
          ->fields(array('timestamp' => REQUEST_TIME))
          ->execute();
      }
    }
  }
}