Fix bad tests