Fixed bug in functions.php