fix tests broken by last commit