Small test tweak