Fix missing function

Bad merge?