Fixed CollectionBase dropping models on .set()