@pv is right I think. Can we use ``sum`` here and merge this?