Take sum after partition rather than aggregate. Small speed-up.