Fix bug with $set with array field