Minor tweak in test