Minor bug fix - Edit button