[refactor]: Cleanup test helper.