update tests code