minor fix to function.php