Update template-tags.php