fix tests that failed from changes in last commit