Fixed the bug in functions.php