fixed bug in functions.php