Fix company test edit