[chore]fix tests to make it pass