Fixed AttributionGenerator tests