Do $update instead of $set