Function fixes