Fix polymorphic edit button