Fixed missing function error from test