Fixed missing function from last commit