Add missing function import