Partial fix to segment collection example