Update Test class