fixup test a bit