Coloque seus snippets em mu-plugins no WordPress em vez do functions.php: o arquivo carrega sempre e sobrevive à atualização do tema.