fixed minor bug in functions.php