Re-add function accidentally removed