Fixes previous bug for edit page.