Minor tweak to functions.php