Fixed bug with setting Edit model values