Can you just use TestHelper.getTestFeature() instead of this method?