fixed unit tests I broke.