Fix attribution rake task