Update functions.php

Bug fix