Fix test tweak