Fix missing function type