Fixed error on edit page