Minor test tweak