Minor tweak to test helper