Add missing primary button class to the edit button