add missing function back