Making update on collection element work