kinda fix up these tests a little