fix works on edit form as well!

Change-Id: I295b885fdb81c270a35668c42647830b4fc22fd5