Minor change to functions.php