Forgot to update scripts outside pipeline usage. Fixed missing imports and changed method name.