Update Test Class