fix code folding on some functions