Use `display_edit_forms` maybe?