Add missing test helper