Fixing little things in some tests that I forgot