[Bug ] Fix collection edit mode [r=amirn]