fix tests broken in last commit