Fixing bug with value not displaying in edit