Fix bug on edit button after ajax submit / front page