Add the function I forgot last commit