Fixing bug in missing function params introduced by COOK-3264