Fix update new class and existing class