fix tests I broke