Removing identity sum nodes, seems to work fine