fixing me some more tests