Fix edit page bug