update tests code.