Fix numpy sum axis notation