Re-add missing function